2021-12-10 12:08:36 索炜达电子 3649
错误调用方法:
正确调用方法:
from sklearn.model_selection import train_test_split
在机器学习中,我们通常将原始数据按照比例分割为“测试集”和“训练集”,经常使用sklearn.cross_validation里的train_test_split模块用来分割数据。
但目前train_test_split已被cross_validation被废弃了
会报错:
ModuleNotFoundError: No module named 'sklearn.cross_validation'
现在改为从 sklearn.model_selection 中调用train_test_split 函数可以解决此问题。