3 mnist 데이터셋에서 X의 형태 바꾸기
28×28 형태의 데이터를 인공지능 모델에 넣으려면 형태를 바꿀 필요가 있습니다. 이제부터 만들 인공 신경망의 입력층에 데이터를 넣을 때는 한 줄로 만들어서 넣어야 하기 때문입니다.
TIP
인공지능을 만들때 항상 입력 데이터를 한 줄로 만들 필요는 없습니다. 인공지능 개발자가 모델을 설계하는 방식에 따라 입력 형태는 바뀔 수 있습니다.
그림 16-9 | 28×28을 1×784로 변환하여 딥러닝 모델에 입력하는 과정
우리가 준비한 데이터의 모습은 28×28의 모습입니다. 이 데이터를 1×784 형태처럼 한 줄로 만든 후 이를 딥러닝 모델에 입력하려 합니다. 지금부터 이 과정을 코딩하여 보겠습니다.
X_train = x_train.reshape(60000, 784) X_test = x_test.reshape(10000, 784) X_train = X_train.astype('float32') X_test = X_test.astype('float32') X_train /= 255 X_test /= 255 print("X Training matrix shape", X_train.shape) print("X Testing matrix shape", X_test.shape)
실행 결과
X Training matrix shape (60000, 784) X Testing matrix shape (10000, 784)