9 잘 예측한 데이터 살펴보기

이제 정확하게 예측한 데이터의 위치와 그렇지 않은 데이터의 위치를 알게 되었습니다. 그렇다면 그 데이터는 어떻게 생겼는지 확인해 보겠습니다. 실제로 우리가 그 결과를 눈으로 살펴볼 수 있도록 matplotlib 라이브러리를 사용해서 화면에 그래프를 출력해 보겠습니다.

plt.figure()
for i in range(9):
    plt.subplot(3,3,i+1)
    correct = correct_indices[i]
    plt.imshow(X_test[correct].reshape(28,28), cmap='gray')
    plt.title("Predicted {}, Class {}".format(predicted_classes[correct], y_test[correct]))
plt.tight_layout()

실행 결과

실행 결과를 보면 총 9개 이미지가 나타나며, 예측한 값과 실제 값이 보입니다. 가장 첫 번째 그림을 보면 예측 값은 7이고 실제 값은 7로 정확하게 예측한 것을 볼 수 있습니다.