모델의 훈련이 끝났으므로, 모델의 정확도를 시각적으로 표현해 봅시다.
코드 5-30 모델의 정확도를 시각적으로 표현
accuracy = history.history['accuracy']
val_accuracy = history.history['val_accuracy']
loss = history.history['loss']
val_loss = history.history['val_loss']
epochs = range(len(accuracy))
plt.plot(epochs, accuracy, label="훈련 데이터셋")
plt.plot(epochs, val_accuracy, label="검증 데이터셋")
plt.legend()
plt.title('정확도')
plt.figure()
plt.plot(epochs, loss, label="훈련 데이터셋")
plt.plot(epochs, val_loss, label="검증 데이터셋")
plt.legend()
plt.title('오차')
다음 그림은 모델의 훈련 결과를 시각적으로 표현한 것입니다.
▲ 그림 5-42 텐서플로 허브를 이용할 경우 정확도 및 손실