我想问一下如何在朴素贝叶斯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
抱歉我的英语不好。
答案 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)
干杯!