더북(TheBook)

incorrect = incorrect_indices[i]

for문의 내부입니다. 앞에서 만든 incorrect_indices 배열에서 첫 번째부터 아홉 번째까지의 값을 반복할 때마다 incorrect 변수에 넣습니다. 첫 번째 반복에서 i의 값은 0입니다. 이때 incorrect_indices 배열의 첫 번째 값이 13이라고 가정해 봅시다. 이는 13번째의 그림을 맞히지 못했다는 의미이고, incorrect 변수에는 13이 들어갑니다.

plt.imshow(X_test[incorrect].reshape(28,28), cmap='gray')

for문의 내부입니다. imshow 함수는 어떤 이미지를 보여줄지에 대한 내용을 담고 있습니다. 첫 번째 반복에서는 X_test 변수에 들어 있는 열세 번째 그림(incorrect 변수에 13가 들어있으니까)을 가져옵니다. 그림의 형태를 바꾸기 위해 reshape(28,28) 함수를 사용하고, 마찬가지로 그림을 회색조로 나타내기 위해 cmap='gray'를 입력합니다.

plt.title("Predicted {}, Class {}".format(predicted_classes[incorrect], y_test[incorrect]))

for문의 내부입니다. 그림 설명을 넣습니다. 예측한 값을 나타내기 위해 Predicted {(값이 들어가는 공간)}에 예측한 결괏값(predicted_classes[incorrect])을, 실제 값을 나타내기 위해 Class {(값이 들어가는 공간)}에 실제 값(y_test[incorrect])을 넣습니다. 이때 format 함수를 사용하여 값을 넣어줍니다.

plt.tight_layout( )

이제 for문을 빠져나와서 화면에 그림을 보여주기 위해 tight_layout 함수를 사용합니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.