훈련과 테스트 데이터가 고르게 분포되어 있는 것을 확인할 수 있습니다.
드롭아웃의 효과를 확인하기 위해 드롭아웃이 적용된 것과 그렇지 않은 것의 모델을 생성합니다.
코드 8-17 드롭아웃을 위한 모델 생성
N_h = 100
model = torch.nn.Sequential(
torch.nn.Linear(1, N_h),
torch.nn.ReLU(),
torch.nn.Linear(N_h, N_h),
torch.nn.ReLU(),
torch.nn.Linear(N_h, 1),
) ------ 드롭아웃이 적용되지 않은 모델
model_dropout = torch.nn.Sequential(
torch.nn.Linear(1, N_h),
torch.nn.Dropout(0.2), ------ 드롭아웃 적용
torch.nn.ReLU(),
torch.nn.Linear(N_h, N_h),
torch.nn.Dropout(0.2),
torch.nn.ReLU(),
torch.nn.Linear(N_h, 1),
) ------ 드롭아웃이 적용된 모델