根据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个。