机器学习训练和测试数据

时间:2019-10-24 19:41:30

标签: python dataframe machine-learning training-data test-data

请问有人知道我该如何使用“ train_test_split()”方法代替手动分割数据?

让我解释一下,实际上我有3个训练文件和2个测试文件,所以我想将训练文件的数据影响到 X_train 和< strong> y_train ,并将测试文件的数据保存到 X_test y_test < / strong>。

谢谢!

1 个答案:

答案 0 :(得分:0)

您可以尝试这样的事情

splitRow = int(x_data.shape[0]*0.99)  #This split data to 99% and 1%

x_train = x_data[0:splitRow ]
y_train = y_data[0:splitRow ,]
x_test =  x_data[splitRow :]
y_test =  y_data[splitRow :]