다음 그림은 개와 고양이에 대한 예측 결과입니다. 노란색은 개와 고양이를 정확하게 예측한 것이고, 빨간색은 예측이 잘못되었음을 의미합니다.

▲ 그림 5-34 개와 고양이에 대한 예측 결과

결과를 살펴보니 예측이 정확하지 않은 것을 확인할 수 있습니다. 물론 훈련 데이터를 더 늘리고, 에포크 횟수도 늘려 보면 더 좋은 결과를 얻을 수 있을 것입니다.

이번에는 텐서플로 허브를 사용하는 예제를 살펴보겠습니다.