다음 그림은 이미지 데이터셋 분류에 대한 예측을 출력한 결과입니다.

▲ 그림 6-7 모델의 예측 결과

텐서보드로 확인했던 것처럼 예측력이 좋지 못한 것을 확인할 수 있습니다.

LeNet-5의 코드를 살펴보았고, 이미지 분류에서 사용되는 또 다른 모델인 AlexNet에 대해 살펴보겠습니다.