如何修复scikit-learn中cross_validation模块的DeprecationWarning？

嘿，这个问题我碰到好多次了，很容易搞定！

你看到的警告已经说得很明白啦：sklearn的cross_validation模块在0.18版本之后就被官方弃用了，所有相关的交叉验证工具都迁移到model_selection模块里了，新模块的接口也更规范统一。

具体修复步骤 #

只需要修改你的导入语句就行，其他代码完全不用变动：

• 原来的导入代码：

  from sklearn.cross_validation import train_test_split
  

• 替换成：

  from sklearn.model_selection import train_test_split
  

修改完再运行代码，那个烦人的弃用警告就会消失啦。而且train_test_split的用法和之前完全一样，你不用改任何调用它的逻辑。

额外小提示 #

不止train_test_split，像原来cross_validation里的KFold、cross_val_score、GridSearchCV这些常用工具，现在也都在model_selection模块里。以后再遇到类似的弃用警告，直接把导入路径里的cross_validation换成model_selection就准没错。

内容的提问来源于stack exchange，提问作者Abdul Basit