【发布时间】:2018-12-27 16:19:33
【问题描述】:
我已经看到了一个如何将 MNIST 数据集拆分为批次并在多个 epoch 中训练它们的示例。我想对 IRIS 数据集做同样的事情,但我不知道如何将 IRIS 拆分为 python 中的批次(或通过 tensorflow 库)。如果有任何帮助,我将不胜感激。
【问题讨论】:
-
如果您想要一个好的答案,需要提供更多信息。
标签: tensorflow epoch training-data
我已经看到了一个如何将 MNIST 数据集拆分为批次并在多个 epoch 中训练它们的示例。我想对 IRIS 数据集做同样的事情,但我不知道如何将 IRIS 拆分为 python 中的批次(或通过 tensorflow 库)。如果有任何帮助,我将不胜感激。
【问题讨论】:
标签: tensorflow epoch training-data
尝试使用此代码,这会将您的数据集分为以下 4 个变量:
from sklearn.model_selection import train_test_split
X_train,X_test,y_train,y_test=train_test_split(dataset_name['data'],dataset_name['target'],random_state=0)
【讨论】: