【发布时间】:2017-02-05 21:56:33
【问题描述】:
给定一个分类问题,有时我们不仅仅预测一个类,还需要返回它是一个类的概率。
即P(y=0|x), P(y=1|x), P(y=2|x), ..., P(y=C|x)
无需构建新的分类器来分别预测 y=0, y=1, y=2... y=C。由于训练 C 分类器(假设 C=100)可能非常慢。
可以做些什么来做到这一点?哪些分类器自然可以轻松给出所有概率(我知道的一个是使用具有 100 个输出节点的神经网络)?但是如果我使用传统的随机森林,我就做不到,对吧?我使用 Python Scikit-Learn 库。
【问题讨论】:
-
您使用的是什么类型的数据?好的、老式的多项式逻辑回归将预测概率。
-
一般。尽管在我的情况下,数字和图像都混合在一起。除了逻辑回归,还有别的吗?
-
错了。 Scikit-Learn 的
RandomForestClassifier具有predict_proba(X)函数,可一次性为您提供所有类别的概率分布。
标签: python machine-learning scikit-learn