train_test_split-不随机,具有原始顺序

时间:2018-12-31 19:45:02

标签: machine-learning split scikit-learn

我想使用train_test_split(X, y, test_size = 0.2),但我不希望数据是随机的-我希望对数据的前80%进行训练,对后20%进行测试。能做到吗?

1 个答案:

答案 0 :(得分:0)

我以为train_test_split在初始混洗关闭时仍在使用随机。这实际上可以通过简单的shuffle=False arg来解决:

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 42,shuffle=False)