且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

在 scikit learn 中使用混淆矩阵作为交叉验证中的评分指标

更新时间:2023-12-03 23:01:34

你可以使用 cross_val_predict(查看 scikit-learn 文档) 而不是 cross_val_score.

You could use cross_val_predict(See the scikit-learn docs) instead of cross_val_score.

而不是做:

from sklearn.model_selection import cross_val_score
scores = cross_val_score(clf, x, y, cv=10)

你可以:

from sklearn.model_selection import cross_val_predict
from sklearn.metrics import confusion_matrix
y_pred = cross_val_predict(clf, x, y, cv=10)
conf_mat = confusion_matrix(y, y_pred)