如何从随机森林中获取树信息,打包'派对'

时间:2014-03-27 16:22:57

标签: r package random-forest party

我正在使用R套餐'派对'。有没有方法可以访问单个树的信息?例如哪些样本(选择哪些特征和时间)用于创建树,这些是OOB样本,树的OOB错误是什么。包'randomForest'似乎提供了部分但不完整的信息。虽然RF是一个黑盒子,但我认为它可能更亮。

1 个答案:

答案 0 :(得分:2)

函数getTree(rfobj,k = 1,labelVar = FALSE)给出每棵树的详细信息。 您可以将整个随机林打印到本地文件,例如,以下代码会将每个树的所有信息打印到.csv文件中。

for (i in 1:RandomForestObject$ntree){
    treeFileName <- paste(outputFileDir,'/',forestName,"_tree_", i, '.csv', sep="")
    tree<-getTree(RandomForestObject,i,labelVar=TRUE)
    write.csv(tree, file=treeFileName)
}

检查属性(RandomForestObj),'err.rate'为每个类提供每棵树的错误率,这也是绘制整个林的错误率​​时显示的内容。