이제 딥러닝을 실행할 차례입니다. 앞서 우리는 학습셋과 테스트셋을 나누는 방법에 대해 알아보았습니다. 이 장에서는 여기에 검증셋을 더해 보겠습니다.

그림 14-1 | 학습셋, 테스트셋, 검증셋

학습이 끝난 모델을 테스트해 보는 것이 테스트셋의 목적이라면, 최적의 학습 파라미터를 찾기 위해 학습 과정에서 사용하는 것이 검증셋입니다. 검증셋을 설정하면 검증셋에 테스트한 결과를 추적하면서 최적의 모델을 만들 수 있습니다. 검증셋은 model.fit() 함수 안에 validation_split이라는 옵션을 주면 만들어집니다. 그림 14-1과 같이 전체의 80%를 학습셋으로 만들고 이 중 25%를 검증셋으로 하면 학습셋:검증셋:테스트셋의 비율이 60:20:20이 됩니다.

전체 코드를 실행하면 다음과 같습니다.