如何在python中预测带有随机森林的多个类?

时间:2016-05-01 09:48:53

标签: python machine-learning scikit-learn

如何在python中预测多个具有随机森林的类?

我熟悉获取概率,但我怎么知道每个概率属于哪个类?

1 个答案:

答案 0 :(得分:2)

根据documentation predict返回一组预测类,您可以使用这些类来获取每个样本最可能的类。

如果您想使用predict_proba来获取所有课程的概率,则需要查看classes_属性。这将按照predict_proba返回其概率的相同顺序列出类。