在sklearn LogisticRegression.predict中缺少概率



我正在尝试将我的数据分为 3 个类别之一。我正在使用 sklearns 逻辑回归,它将一个单词作为输入到我的测试数据的 vec 矢量化形式。预测方法返回预测,但它没有给出关于模型在分类中的概率或置信度的数字是否正确?有人可以告诉我如何获得这些信息吗?

如果你仔细看看文档,还有另一个函数, predict_proba ,来获取概率。

所以你只需要做:

clf = LogisticRegression()
clf.predict_proba(X_input)

最新更新