可视化随机森林回归树中的一棵树
时间: 2023-08-13 10:07:41 浏览: 90
要可视化随机森林中的一棵回归树,可以先使用随机森林模型来拟合数据集,然后从中选择一棵树来进行可视化。
以下是一个示例代码:
```python
from sklearn.ensemble import RandomForestRegressor
from sklearn.datasets import load_boston
from sklearn.tree import export_graphviz
import pydotplus
from IPython.display import Image
from graphviz import Source
# 加载波士顿房价数据集
boston = load_boston()
# 构建随机森林模型
rf = RandomForestRegressor(n_estimators=100, random_state=0)
rf.fit(boston.data, boston.target)
# 选择一棵树进行可视化
tree = rf.estimators_[0]
# 可视化回归树
dot_data = export_graphviz(tree, out_file=None, feature_names=boston.feature_names)
graph = pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png())
```
此代码将生成一张PNG格式的回归树图像,并使用Jupyter Notebook显示该图像。
如果您想直接在Python中显示图像,可以使用graphviz的Source类:
```python
graph = Source(dot_data)
graph
```
这将在Notebook中显示图像。
阅读全文