如何从随机森林中的每棵树获取节点值?

时间:2019-05-04 16:51:03

标签: python-3.x scikit-learn random-forest

我正在为多输出回归问题生成随机森林。最后,预测方法给出的结果是给定x的结果,该结果是森林中不同树木生成的所有不同结果的平均值。我正在寻找这些发束以numpy格式生成的所有“ y”(可能是)?我不确定scikit-learn的RandomForestRegressor中是否有可用的方法。

这是我的树的样子: enter image description here

estimator = model2.estimators_[23][9]
from sklearn.externals.six import StringIO  
from IPython.display import Image  
from sklearn.tree import export_graphviz
import pydotplus
dot_data = StringIO()
export_graphviz(estimator, out_file=dot_data,  
                filled=True, rounded=True,
                special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())

我期待

  

[2,3,3]

但我得到的是

  

2.6

0 个答案:

没有答案