sklearn 模型评估

Posted 2022-11-07 hichens

原文链接

http://d0evi1.com/sklearn/model_evaluation/

预测值：pred

真实值：y_test

 

直接用平均值

mean(pred == y_test)

 

准确得分

from sklearn.metrics import accuracy_score

accuracy_score(pred, y_test)

accuracy_score(pred, y_test, normalize=False)

 

混淆矩阵

from sklearn.metrics import confusion_matrix

confusion_matrix(pred, y_test)

 

 

汉明
from sklearn.metrics import hamming_loss

hamming_loss(pred, y_test)