zl程序教程

您现在的位置是:首页 >  其他

当前栏目

ImportError: cannot import name ‘cross_validation‘ 解决方法

2023-03-20 15:33:45 时间

文章目录

一、问题

在学习机器学习分类算法 KNN 时,使用交叉验证时,调用了 cross_validation 函数,导入时报错。 查阅资料和官方文档后发现:sklearn在 0.02 版本后改变了 cross_validation 函数 https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.cross_validate.html

二、解决方法

cross_val_score、cross_validate均可用于交叉验证,返回值是scores,即每次交叉验证的得分。

from sklearn.model_selection import cross_validate

测试如下:

from sklearn.model_selection import cross_validate
from sklearn.model_selection import cross_val_score
from sklearn import datasets
from sklearn import svm

iris = datasets.load_iris()
clf = svm.SVC(kernel='linear', C=1)
X = iris.data
y = iris.target
# 交叉验证
scores = cross_val_score(clf, X, y, cv=3)
print(scores)
# print(scores['test_score'])

结果如下:

可以用于测试交叉验证啦。