더북(TheBook)

2.8.3 모델 훈련

다음으로 다중 레이어 퍼셉트론을 훈련시킨다. fit 함수를 호출하면 된다. 훈련 반복 횟수는 200회로 하자.

# 200회 반복 훈련시킨다
model.fit(X_train, y_train, epochs=200)

▲ 그림 2-16 모델을 200회 반복 훈련

결과에서 볼 수 있듯이 각 에폭(epoch)별로 훈련 데이터에 따라 가중치와 편향을 계속 갱신하면서 손실이 점차 줄어들고 정확도가 증가했다. 하지만 그림 2-16의 정확도는 어디까지나 훈련 데이터셋으로 계산한 것이다. 다음 2.9절에서 따로 빼둔 테스트 데이터셋과 추가 지표를 사용해 모델 성능을 최종 평가하는 방법을 알아보자.

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