如何为深度学习创建数据集？

Question

我想和Keras一起训练CNN。我了解x_train包含数据功能和y_train标签。

我尝试了test_train_split(x_train,test_size=0.2,train_size=0.8)。

如何创建y_train？

Answer 1

你走了：

from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.4, random_state=None, shuffle=False)

如果是Timeseries shuffle = False。 test_size是您数据的40％。所以你要训练60％