封装的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)的顺序?

http://img.mukewang.com/szimg/5aef0fda0001d73b12070350.jpg

http://img.mukewang.com/szimg/5aef0fdb00018d0913520461.jpg


写回答

2回答

liuyubobobo

2018-05-06

课程封装的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,没有问题,但是在具体调用的时候需要修改。

0
0

慕田峪210210

提问者

2018-05-06

好的,谢谢波波老师

0
0

Python3入门机器学习 经典算法与应用  

Python3+sklearn,兼顾原理、算法底层实现和框架使用。

5839 学习 · 2437 问题

查看课程