RepeatDataset怎么用sklearn.model_selection.train_test_split来进行训练集和测试集的划分呢?

tensorflow.python.data.ops.dataset_ops.RepeatDataset

这个对象怎么用sklearn.model_selection.train_test_split来进行训练集和测试集的划分呢?

先把数据集转成nd.array或者dataframe形式吧