train_test_split 不打乱顺序,如何设定参数

窦弘义
2023-12-01

使用 scikit-learn 的 train_test_split 函数,可以通过设置 shuffle 参数为 False 来保持原数据的顺序不变。例如:

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

其中 X 和 y 分别为训练数据和标签,test_size 参数表示测试集所占比例。

 类似资料: