根据Tensorflow Dataset
docs,百分比拆分是可能的,例如
first_10_percent= tfds.Split.TRAIN.subsplit(tfds.percent[:10])
split如示例所示,更改列表时,您的代码将起作用:
(training_set, validation_set), dataset_info = tfds.load('tf_flowers',split=[ tfds.Split.TRAIN.subsplit(tfds.percent[:70]), tfds.Split.TRAIN.subsplit(tfds.percent[70:])],with_info=True,as_supervised=True,)
使用上面的代码,
training_set有2590个条目,而
validation_set有1080个。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)