모델을 컴파일하고 훈련시킵니다. 단 조기 종료의 효과를 알아보기 위해 이번에는 조기 종료 코드는 추가하지 않습니다.
코드 8-31 조기 종료가 추가되지 않은 모델 훈련
model.compile(loss='binary_crossentropy',
optimizer='adam',
metrics=['accuracy'])
model.fit(x_train, y_train,
batch_size=batch_size,
epochs=epochs,
validation_split=0.2,
callbacks=[cp_callback])
다음은 조기 종료가 추가되지 않은 모델 훈련 결과입니다.
Epoch 1/300
624/625 [============================>.] - ETA: 0s - loss: 0.4360 - accuracy: 0.7764
Epoch 00001: saving model to training_1\cp.ckpt
...(중간 생략)...
Epoch 300/300
625/625 [==============================] - ETA: 0s - loss: 0.0014 - accuracy: 0.9996
Epoch 00300: saving model to training_1\cp.ckpt
INFO:tensorflow:Assets written to: training_1\cp.ckpt\assets
625/625 [==============================] - 33s 52ms/step - loss: 0.0014 - accuracy: 0.9996 - val_loss: 1.8912 - val_accuracy: 0.8754
<tensorflow.python.keras.callbacks.History at 0x26287c0eba8>