评估指标与评分

2019-01-23  本文已影响0人  dreampai

一、二分类指标

1、混淆矩阵

image.png

二、多分类指标

对于多分类问中的不均衡数据集,最常用的指标就是多分类版本的 f-分数。f-分数背后的想法是:对每一个类别计算一个二分类 f-分数,其中该类别是正类,其他所有类别组成反类。然后,使用以下策略对这些按类别 f-分数进行平均:

from sklearn.datasets import load_digits
from sklearn.metrics import accuracy_score
from sklearn.metrics import confusion_matrix
from sklearn.metrics import classification_report
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import f1_score
digits=load_digits()
X_train,X_test,y_train,y_test=train_test_split(digits.data,digits.target,random_state=0)
lr=LogisticRegression()
lr.fit(X_train,y_train)
pred=lr.predict(X_test)
print('Accuracy:{:.3f}'.format(accuracy_score(y_test,pred)))
print('Confusion matarix:\n{}'.format(confusion_matrix(y_test,pred)))
print(classification_report(y_test,pred))
print('Micro average f1 score:{:.3f}'.format(f1_score(y_test,pred,average='micro')))
print('Macro average f1 score:{:.3f}'.format(f1_score(y_test,pred,average='macro')))

三、回归指标

R2 是评估回归模型更直观的指标。

上一篇下一篇

猜你喜欢

热点阅读