ModuleNotFoundError: No module named 'sklearn.cross_validation'

sklearn切分数据集

最新推荐文章于 2024-03-21 14:40:34 发布

原创最新推荐文章于 2024-03-21 14:40:34 发布 · 799 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习

AI 同时被 3 个专栏收录

6 篇文章

订阅专栏

人工智能

5 篇文章

订阅专栏

Python

4 篇文章

订阅专栏

在机器学习中，使用sklearn进行数据集切分时，若遇到ModuleNotFoundError错误，提示找不到'sklearn.cross_validation'模块，这是因为从sklearn0.18版本开始，此模块已被废弃。解决方案是将'cross_validation'替换为'model_selection'，并使用train_test_split进行数据集分割。

ModuleNotFoundError: No module named 'sklearn.cross_validation'

在机器学习中，我们通常将原始数据按照比例分割为“训练集”和“测试集”，通常使用sklearn.cross_validation里的train_test_split模块用来分割数据

但在sklearn 0.18及以上的版本中，出现了sklearn.cross_validation无法导入的情况，原因是新版本中此包被废弃

from sklearn.cross_validation import train_test_split

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-14-d05cc6ee7707> in <module>
----> 1 from sklearn.cross_validation import train_test_split

ModuleNotFoundError: No module named 'sklearn.cross_validation'

ModuleNotFoundError

只需将 cross_validation 改为 model_selection 即可

from sklearn.model_selection import train_test_split