标签: python machine-learning scikit-learn
如何在python中预测多个具有随机森林的类?
我熟悉获取概率,但我怎么知道每个概率属于哪个类?
答案 0 :(得分:2)
根据documentation predict返回一组预测类,您可以使用这些类来获取每个样本最可能的类。
predict
如果您想使用predict_proba来获取所有课程的概率,则需要查看classes_属性。这将按照predict_proba返回其概率的相同顺序列出类。
predict_proba
classes_