sklearn中从sklearn.cross_validation 导入 train_test_split报错No module named ‘sklearn.cross_validation‘

问题描述

从sklearn.cross_validation里选择导入train_test_split用于数据分割时，报错No module named ‘sklearn.cross_validation’

解决办法

新版本的sklearn中导入train_test_split用于数据分割是这样的：

from sklearn.model_selection import train_test_split

有的帖子说还要加上from sklearn.model_selection import KFold，实测目前不加这一句没啥问题。只用将旧版的sklearn.cross_validation更换为model_selection即可