더북(TheBook)

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)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.