在朴素贝叶斯python代码中添加混淆矩阵

时间:2017-11-13 04:13:54

标签: python naivebayes confusion-matrix

我想问一下如何在朴素贝叶斯python代码中添加混淆矩阵。 我从中获得了朴素的贝叶斯代码   naive bayes for iris data

我需要一些改变来添加混淆矩阵。 我有混淆矩阵的代码,但我不知道在我的代码上放哪个,我不知道要替换什么???什么。

y_actu = pd.Series(???, name='Actual')
y_pred = pd.Series(???, name='Predicted')
df_confusion = pd.crosstab(y_actu, y_pred, rownames=['Actual'], colnames=['Predicted'], margins=True)

print df_confusion

抱歉我的英语不好。

1 个答案:

答案 0 :(得分:0)

我建议您使用sklearn' confusion_matrix函数

from sklearn.metrics import  confusion_matrix

y_actu = pd.Series(???, name='Actual')
y_pred = pd.Series(???, name='Predicted')

confusionMatrix = confusion_matrix(y_actu, y_pred)
print(confusionMatrix)

Documentation here

干杯!