더북(TheBook)

2.2.4 모델 훈련

앞서 만들어 둔 데이터로 모형을 학습시킵니다. 이때 학습을 시킨다는 것은 y=wx+b라는 매핑 함수에서 wb의 적절한 값을 찾는다는 의미입니다. wb에 임의의 값을 적용하여 시작하며, 모델에 데이터를 입력하면서 오차를 구하게 됩니다. 이때 오차가 줄어드는 방향으로 파라미터를 수정합니다.

다음은 모델을 훈련시키는 예시 코드입니다.

model.fit에 대한 파라미터는 다음과 같습니다.

ⓐ 입력 데이터

ⓑ 결과(label) 데이터

ⓒ 학습 데이터 반복 횟수

ⓓ 한 번에 학습할 때 사용하는 데이터 개수

ⓔ 검증 데이터(validation data)를 나타내는 것으로, 검증 데이터를 사용하면 각 에포크마다 검증 데이터의 정확도도 함께 출력됩니다. 하지만 이 정확도는 훈련이 잘 되고 있는지 보여 줄 뿐이며, 실제로 모델이 검증 데이터를 학습하지는 않습니다.

ⓕ 학습 진행 상황을 보여 줄지 지정하는 것으로, 1 값으로 설정하면 학습 진행 상황을 볼 수 있습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.