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