python - KeyError：“无效的拆分火车[：80％]。可用的拆分是：['train']”

Question

在链接：https ://www.tensorflow.org/tutorials/images/transfer_learning

(raw_train, raw_validation, raw_test), metadata = tfds.load( 'cats_vs_dogs', split=['train[:80%]', 'train[80%:90%]', 'train[90%:]'], with_info=True, as_supervised=True, )

为什么会有错误

KeyError: "Invalid split train[:80%]. Available splits are: ['train']"

标签： pythontensorflowgoogle-colaboratory

Answer 1

试试这段代码我们可以在 TF 2 中将其拆分为：

(raw_train, raw_validation, raw_test), metadata = tfds.load(
'cats_vs_dogs',
split=[
   tfds.Split.TRAIN.subsplit(tfds.percent[:80]),
   tfds.Split.TRAIN.subsplit(tfds.percent[80:90]),
   tfds.Split.TRAIN.subsplit(tfds.percent[90:])
],
with_info=True,
as_supervised=True,
)