AttributeError：类型对象“Split”没有属性“Train”

Question

我正在遵循使用 Google colab 的简单教程。

(train_data,validation_data),test_data=tfds.load(name='imdb_reviews',
          split=((tfds.Split.Train.subsplit([6,4])),tfds.Split.Test),
          as_supervised=True)

编写此代码后，我收到如下错误：

AttributeError                            Traceback (most recent call last)
<ipython-input-10-facc01319dcb> in <module>()


         1 (train_data,validation_data),test_data=tfds.load(name='imdb_reviews',
    ----> 2           split=((tfds.Split.Train.subsplit([6,4])),tfds.Split.Test),
          3           as_supervised=True)

AttributeError: type object 'Split' has no attribute 'Train'

Answer 1

.subsplit是已弃用 API 的一部分，因此无法再使用。

相反，您应该使用新的 subsplit API： https: //www.tensorflow.org/datasets/splits

ds1, ds2 = tfds.load(..., split=[ 'train[:50%]', 'train[50%:]'])

根据1998年的问题。