如何在python中用随机森林预测多个类?
Posted
技术标签:
【中文标题】如何在python中用随机森林预测多个类?【英文标题】:How to predict more than one class with random forest in python? 【发布时间】:2016-08-26 04:54:40 【问题描述】:如何在 python 中使用随机森林预测多个类?
我对get概率很熟悉,但是我怎么知道每个概率属于哪个类呢?
【问题讨论】:
【参考方案1】:根据documentation predict
返回一个预测类数组,您可以使用它来获取每个样本最可能的类。
如果要使用predict_proba
来获取所有类的概率,则需要查看classes_
属性。这会按照 predict_proba
返回其概率的相同顺序列出类。
【讨论】:
你如何让它返回多个类? 有没有办法获得多个值的回报。对于我的问题,我可以解析 classes_,但这很慢。 我不确定我是否理解。为什么您认为解析classes_
会很慢?
如果我想找到 n 个最有可能的类。我需要在get_proba矩阵中找到n个最大值,这涉及到排序。
当然。但是您(可能)没有那么多课程。并且必须有人进行排序,即使提供了一个功能可以为您完成排序。我建议您在开始优化之前先使用简单的方法来确定排序是否真的是您的瓶颈。以上是关于如何在python中用随机森林预测多个类?的主要内容,如果未能解决你的问题,请参考以下文章
如何在 Python scikit-learn 中输出随机森林中每棵树的回归预测?