封装的model_selection中return 的顺序问题
来源:4-3 训练数据集,测试数据集
慕田峪210210
2018-05-06
老师您好,为啥我封装的model_selection中return 的顺序是X_train,y_train,X_test,y_test,然而调用的时候还得按照X_train,X_test,y_train,y_test = train_test_split(X,y)的顺序?
写回答
2回答
-
课程封装的model_selection中train_test_split的顺序是X_train, X_test, y_train, y_test。具体可以参见课程官方代码:https://github.com/liuyubobobo/Play-with-Machine-Learning-Algorithms/blob/master/04-kNN/03-Test-Our-Algorithm/playML/model_selection.py
如果你想修改成返回顺序是X_train, y_train, X_test, y_test,没有问题,但是在具体调用的时候需要修改。
00 -
慕田峪210210
提问者
2018-05-06
好的,谢谢波波老师
00
相似问题