더북(TheBook)

케라스에서 제공하는 데이터셋 사용

이때는 텐서플로에서 제공하는 케라스 모듈을 사용합니다.

import tensorflow as tf
data_train, data_test = tf.keras.datasets.mnist.load_data() ------ 케라스를 이용하여 MNIST 데이터셋을 내려받기
(images_train, labels_train) = data_train ------ 훈련 데이터셋 정의
(images_test, labels_test) = data_test ------ 검증 데이터셋 정의

 

 

인터넷에서 데이터셋을 로컬 컴퓨터에 내려받아 사용

이때는 텐서플로에서 제공하는 tf.keras.utils.get_file 메서드를 사용합니다.

import tensorflow as tf
url = 'https://storage.googleapis.com/download.tensorflow.org/data/illiad/butler.txt'
text_path = tf.keras.utils.get_file("butler.txt", origin=url)
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.