我正在尝试将我的数据分为 3 个类别之一。我正在使用 sklearns 逻辑回归,它将一个单词作为输入到我的测试数据的 vec 矢量化形式。预测方法返回预测,但它没有给出关于模型在分类中的概率或置信度的数字是否正确?有人可以告诉我如何获得这些信息吗?
如果你仔细看看文档,还有另一个函数, predict_proba
,来获取概率。
所以你只需要做:
clf = LogisticRegression()
clf.predict_proba(X_input)