【发布时间】:2019-11-20 16:51:31
【问题描述】:
我正在尝试将我的数据分类为 3 个类别之一。我正在使用 sklearns 逻辑回归,它将一个单词作为我的测试数据的 vec 向量化形式的输入。 predict 方法返回预测,但它没有给出模型对分类正确的概率或置信度的数字?有人可以告诉我如何获得这些信息吗?
【问题讨论】:
标签: python machine-learning scikit-learn logistic-regression
我正在尝试将我的数据分类为 3 个类别之一。我正在使用 sklearns 逻辑回归,它将一个单词作为我的测试数据的 vec 向量化形式的输入。 predict 方法返回预测,但它没有给出模型对分类正确的概率或置信度的数字?有人可以告诉我如何获得这些信息吗?
【问题讨论】:
标签: python machine-learning scikit-learn logistic-regression
如果您仔细查看documentation,还有另一个函数predict_proba 可以获取概率。
所以你只需要这样做:
clf = LogisticRegression()
clf.predict_proba(X_input)
【讨论】: