分享

数据分享|Python在Scikit-Learn可视化随机森林中的决策树分析房价数据

 拓端数据 2022-06-06 发布于上海

原文链接:http:///?p=27050 

随机森林是决策树的集合。在这篇文章中,我将向您展示如何从随机森林中可视化决策树。

首先让我们在房价数据集查看文末了解数据获取方式上训练随机森林模型。

加载数据并训练随机森林。

X = pd.DataFrame(datacolumns=feature_names)

让我们将森林中的树数设置为 100:

RandomForestRegressor(n_estimators=100)

决策树存储在 模型list 中的 estimators_ 属性中 rf 。我们可以检查列表的长度,它应该等于 n_estiamtors 值。

len(estimators_)>>> 100

我们可以从随机森林中绘制第一棵决策树( 0 列表中有索引):

plot\_tree(rf.estimators\_\[0\])

这棵树太大,无法在一个图中将其可视化。


    转藏 分享 献花(0

    0条评论

    发表

    请遵守用户 评论公约

    类似文章 更多