您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

使用tensorflow_datasets.load(TF 2.1)拆分训练数据以进行训练和验证

使用tensorflow_datasets.load(TF 2.1)拆分训练数据以进行训练和验证

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

其他 2022/1/1 18:26:56 有476人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶