标签： python machine-learning scikit-learn logistic-regression
我正在尝试将数据分类为3类之一。我正在使用sklearns logistic回归，它以一个单词作为我的测试数据的vec矢量化形式作为输入。预测方法返回预测，但是没有给出有关模型在分类正确的概率或置信度的数字？有人可以告诉我如何获取该信息吗？
答案 0 :(得分：2)
如果您仔细观察documentation，还有另一个函数predict_proba来获得概率。
predict_proba
所以您只需要这样做：
clf = LogisticRegression() clf.predict_proba(X_input)