在sklearn中获取MultinomialNB中评分函数的召回率和精度

Question

我想获得预测的召回率和精确度，但我很困惑。 sklearn 中的 MultinomialNB 中的这个函数“score”无法返回预测列表或帮助我达到精度和召回率的东西。例如：

from sklearn.naive_bayes import MultinomialNB
model = MultinomialNB()
model.fit(x_train, y_train)
model.score(x_test, y_test)

这里我有一个要预测的模型，但 MultinomialNB() 没有提供精度或召回率或预测列表等的函数。

Answer 1

我也没有找到它，但这应该可以解决问题：

这应该可以完成工作：

from sklearn.metrics import classification_report

classification_report(y_test, model.predict(x_test))

或者你可以一一导入

from sklearn.metrics import recall_score, f1_score, precision_score

recall = recall_score(y_test, model.predict(x_test))
...
...