[Tensorflow] tensorflow mnist 데이터셋 로드

 

 

 

텐서플로우에서 mnist 데이터셋을 사용하기 위해 사용했던 기존의 명령어인 다음의 코드로 동작하지 않는 다는 문제점이 있다.

 

from tensorflow.examples.tutorials.mnist import input_data

이에 대한 이유로는 텐서플로우 github를 확인하면  tf.contrib.learn.dataset으로 부터 가져오는 것을 더이상 사용하지 말 것을 당부하고 있으며 scikits.learn과 같은 다른 모듈을 통해 데이터 셋을 불러올 것을 권장하고 있다.

 

'tf.contrib.learn.datasets' is deprecated. We are adding ready to use datasets to tensorflow/models. Many smaller datasets are available from other sources, such as 'scikits.learn'. Some Python processing may have to be written, but this is straightforward to implement using the standard modules.

 

이를 위해 다음과 같은 방법으로 데이터셋을 불러올 수 있다.

 

1. 데이터셋 로드

import tensorflow as tf
data_train, data_test = tf.keras.datasets.mnist.load_data()

 

2. 데이터셋 다운로드

import tensorflow as tf
tf.keras.datasets.mnist.load_data(path='mnist.npz')