더북(TheBook)

fashion_mnist 데이터셋을 내려받은 후 메모리로 로딩하고 훈련과 검증 데이터셋으로 분리합니다.

코드 13-9 fashion_mnist 데이터셋 로딩 및 데이터셋 분리

(train_images, train_lebels), (test_images, test_lebels) =
                               tf.keras.datasets.fashion_mnist.load_data() ------ 훈련과 검증 데이터셋으로 분리 
train_images = train_images.reshape(train_images.shape[0], 28, 28,
                                     1).astype("float32") / 255.0
test_images = test_images.reshape(test_images.shape[0], 28, 28, 1).astype("float32") / 255.0

train_dataset = (
    tf.data.Dataset.from_tensor_slices(train_images)
    .shuffle(60000)
    .batch(512)
) ------ 훈련 데이터셋 준비
test_dataset = (
    tf.data.Dataset.from_tensor_slices(test_images)
    .shuffle(10000)
    .batch(512)
) ------ 검증 데이터셋 준비
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.