이번에는 오차 정보를 그래프로 살펴보겠습니다.

코드 5-27 훈련 데이터의 오차에 대한 그래프 확인

plt.plot(train_loss_hist)
plt.show()

그림 5-38은 훈련 데이터에 대해 에포크가 진행될 때마다 오차를 출력한 결과입니다. 에포크가 진행될수록 오차가 낮아지고 있기 때문에 학습이 잘되었다고 할 수 있습니다.

▲ 그림 5-38 훈련 데이터에 대한 오차

단순히 정확도와 오차 정보를 확인하는 것에 머무르지 않고, 실제로 데이터를 잘 예측하는지 살펴보겠습니다.