原文链接:http:///?p=27050随机森林是决策树的集合。在这篇文章中,我将向您展示如何从随机森林中可视化决策树。 首先让我们在房价数据集(查看文末了解数据获取方式)上训练随机森林模型。 加载数据并训练随机森林。 X = pd.DataFrame(data, columns=feature_names) 让我们将森林中的树数设置为 100: RandomForestRegressor(n_estimators=100) 决策树存储在 模型 len(estimators_)>>> 100 我们可以从随机森林中绘制第一棵决策树( plot\_tree(rf.estimators\_\[0\]) 这棵树太大,无法在一个图中将其可视化。 |
|