tf.keras.datasets

tf.keras.dataset 모듈은 TensorFlow에서 사용할 수 있는 데이터셋의 모음입니다.

아래의 7개의 데이터셋 모듈을 포함합니다.

  • boston_housing

  • cifar10

  • cifar100

  • fashion_mnist

  • imdb

  • mnist

  • reuters



예제

import tensorflow as tf

keras = tf.keras

(x_train, y_train), (x_test, y_test) = keras.datasets.mnist.load_data()
assert x_train.shape == (60000, 28, 28)
assert x_test.shape == (10000, 28, 28)
assert y_train.shape == (60000,)
assert y_test.shape == (10000,)

tf.keras.datasets.mnist 모듈은 load_data() 함수를 포함합니다.

load_data() 함수를 처음으로 호출하면 MNIST 데이터셋을 다운로드합니다.

MNIST 손글씨 이미지 분류하기 페이지를 참고하세요.



이전글/다음글


공유하기