机器学习张量流数据集分割根本不运行

Question

我想使用张量流将数据集拆分为我的 cnn 图像分类项目的训练和测试部分。

我正在使用的命令未在我的环境中运行。我已多次重新启动内核，并确保

train_size

、

val_size

和

test_size

是 整数。我使用的数据对象是

tf.data.Dataset object

所以这不应该是一个问题。这些批次确实存在。

这就是开头提到的命令：

train_data = data.take(train_size)
val_data = data.skip(train_size).take(val_size)
test_data = data.skip(train_size + val_size)