将虹膜数据集分成批次并在各个时期进行训练

时间:2018-07-19 13:38:14

标签: tensorflow epoch training-data

我已经看到了一个示例,该示例如何将MNIST数据集拆分为多个批次,并在几个时期内对其进行训练。我想对IRIS数据集执行相同的操作,但是我不知道如何在python中(或通过tensorflow库)将IRIS拆分为批处理。感谢您的帮助。

1 个答案:

答案 0 :(得分:0)

尝试使用此代码,这会将您的数据集分为以下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)