用python求二元分类的混淆矩阵

sklearn里可以直接计算混淆矩阵sklearn.metrics.confusion_matrix 

from sklearn.metrics import confusion_matrix
confusion_matrix(y_true, y_pred)

也可以通过confusion_matrix函数直接得到TN, FP, FN, TP

tn, fp, fn, tp = confusion_matrix(y_true, y_pred).ravel()

confusion matrix本质上就是列联表，pandas里的crosstab可以得到列联表。参考下面的代码

>> results = pd.DataFrame()
>> results['True'] = [1, 1, 1, 1, 1, 0, 0, 0]
>> results['Pred'] = [1, 1, 0, 0, 1, 1, 0, 0]

# 用pd.crosstab可以得到confusion matrix
>> pd.crosstab(results['True'], results['Pred'])

Pred	0	1
True		
0	2	1
1	2	3