如何绘制从使用 R 中的“caret”包创建的随机森林中选择的树
Posted
技术标签:
【中文标题】如何绘制从使用 R 中的“caret”包创建的随机森林中选择的树【英文标题】:How can I plot a tree selected from the random forest created using "caret" package in R 【发布时间】:2018-12-09 17:36:34 【问题描述】:我是 R 的新手,我需要知道如何绘制从 随机森林 训练模型中选择的树,该训练模型是使用 中的 train () 函数创建的插入符号包。
首先,我使用了一个训练数据集,使用 train() 函数创建了一个随机森林的拟合模型。创建的随机森林包含大约 500 棵树。是否有任何方法可以创建选定树的图?
谢谢。
【问题讨论】:
【参考方案1】:据我所知,randomForest
包没有任何内置功能来绘制单个树。您可以使用 getTree()
函数提取树,但没有提供任何东西来绘制/可视化它。这个问题可能是重复的,因为快速搜索产生了其他人用来从随机森林中提取树木的方法
here 和 here 和 here
【讨论】:
【参考方案2】:CRAN 包party
提供了一个名为prettyTree
的方法。
看here
【讨论】:
以上是关于如何绘制从使用 R 中的“caret”包创建的随机森林中选择的树的主要内容,如果未能解决你的问题,请参考以下文章
R语言使用caret包构建gbdt模型(随机梯度提升树Stochastic Gradient Boosting )构建回归模型通过method参数指定算法名称
R语言使用caret包构建随机森林模型(random forest)构建回归模型通过method参数指定算法名称通过ntree参数指定随机森林中树的个数
R语言使用caret包构建正则化的随机森林模型(Regularized Random Forest )构建回归模型通过method参数指定算法名称通过trainControl函数控制训练过程