Deep Learning & Machine Learning/강좌&예제 코드
Tensorflow Dataset 일부만 사용하기
webnautes
2021. 12. 8. 21:43
반응형
Tensorflow Dataset의 일부만 사용하는 예제입니다.
2021. 12. 6 - 최초작성
2021. 12. 8 - 최종작성
import tensorflow_datasets as tfds
# tensorflow dataset 'minst'의 train과 test를 각각 전체를 사용합니다.
ds = tfds.load('mnist', split=['train', 'test'])
print(len(ds[0]), len(ds[1]))
# tensorflow dataset 'minst'의 train과 test를 각각 10%씩 사용합니다.
ds = tfds.load('mnist', split=['train[:10%]', 'test[:10%]'])
print(len(ds[0]), len(ds[1]))
# tensorflow dataset 'minst'의 train을 다른 범위로 각각 10%씩 사용합니다.
ds = tfds.load('mnist', split=['train[10%:20%]', 'train[:10%]'])
print(len(ds[0]), len(ds[1]))
# tensorflow dataset 'minst'의 train의 앞부분 80%는 train으로 나머지 뒷부분 20%는 validation으로 사용하고 test는 전체를 사용합니다.
ds = tfds.load('mnist', split=['train[:80%]', 'train[80%:]', 'test'])
print(len(ds[0]), len(ds[1]), len(ds[2]))
실행결과입니다.
60000 10000
6000 1000
6000 6000
48000 12000 10000
반응형