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

使用tensorflow_datasets.load(TF 2.1)拆分训练数据以进行训练和验证,第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个。



欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5508458.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-13
下一篇 2022-12-13

发表评论

登录后才能评论

评论列表(0条)

保存