4 mnist 데이터셋에서 Y의 형태 바꾸기

지금부터 y_train 데이터와 y_test 데이터의 형태를 바꾸겠습니다. 인공지능이 분류를 잘 할 수 있도록 하기 위해서입니다. 우리가 만들고 있는 인공지능은 이미지를 0~9 사이의 숫자로 분류하는 인공지능입니다. 이를 다시 살펴보면 인공지능은 이미지가 가진 숫자의 특성, 즉 “이 숫자는 3이고 2보다 1 더 큰 수이다.”와 같은 특성을 알 필요는 없습니다. 우리가 만드는 인공지능의 목표는 3과 2를 잘 구분하면 되는 것이죠.

그림 16-11 | 숫자의 특성을 학습하는 인공지능

그러므로 이미지의 레이블(정답)을 인공지능에게 0, 1, 2, 3, 4..와 같이 숫자로 알려주는 것이 아니라 더 잘 구분할 수 있는 방법으로 알려줄 필요가 있습니다. 바로 0은 0이라는 숫자의 의미보다 인공지능이 구분할 10개의 숫자 중 첫 번째 숫자로, 1은 1이라는 숫자의 의미보다 두 번째 숫자로 말해 주는 것이죠.