使用RandomForestClassifier.predict_proba和RandomForestRegressor.predict

Question

我有一个数据集，包括一个特征向量和一个目标-- 1.0或0.0 (代表两个类)。如果我安装了一个RandomForestRegressor并调用它的predict函数，它是否等同于使用RandomForestClassifier.predict_proba()

换句话说，如果目标是1.0或0.0，那么RandomForestRegressor输出概率是多少？

我想是的，而且我得到的结果也表明了这一点，但我想得到第二种意见.

谢谢Weasel

Answer 1

根据所处理的不同任务，这些任务之间在概念上有重大差异：

回归:连续(实值)目标变量。

分类:离散目标变量(类)。

对于一般的分类方法，术语probability of observation being class X可能没有定义，因为某些分类方法，例如knn，不处理概率。

然而，对于随机森林(以及其他一些分类方法)，分类被简化为类概率的回归。然后将预测类作为计算“概率”的论证。在您的例子中，您输入相同的输入，得到相同的结果。是的，可以将RandomForestRegressor返回的值视为概率。