※ 引述《SFMAndroid (安卓发送)》之铭言：
: 承前篇
: 目前成功用plot.getTree得到结果了
: #code
: library(tree)
: library(reprtree)
: library(plotrix)
: set.seed(100)
: jepg("~//tree.jpeg", width=1300, height=800)
: reprtree:::plot.getTree(rf, k=10, depth=0)
: dev.off()
我看了一下，原作者好像写死了那个部分的参数(cex = 0.8)
所以你需要用git clone把package download之后自己做一些修改
plor.reptree.R的 labelBG加上一个input, cex.label
boxed.labels里面的cex引数原本等于0.8要改成 cex = cex.label
像下面这样：
labelBG <- function(tr, cex.label = 0.8){
## 中间省略
boxed.labels(xy$x[ind],xy$y[ind]+0.5*charht, rows[ind] , border=F,
bg='white', cex=cex.label, xpad=0.5, ypad=1)
}
plot.getTree.R的plot.getTree加上一个input, cex.label
然后改倒数第三行的labelBG，改成labelBG(x, cex.label)
像这样
plot.getTree <- function(rforest=NULL,tr=NULL,k=1, depth=0,main=NULL,
cex.label = 0.8, ...){
## 中间省略
labelBG(x, cex.label)
labelYN(x)
title(main=main)
}
你接下来这样就可以work了
library(reprtree)
set.seed(100)
data <- data.frame(y=sample(c("好", "不好"), 100, replace=TRUE),
x1=sample(c("学生", "老师", "教官"), 100, replace=TRUE),
x2=sample(c("20岁以下", "20~30岁", "30岁以上"), 100, replace=TRUE))
rf <- randomForest(y~., data=data, ntree=10, proximity=TRUE)
reprtree:::plot.getTree(rf, k=10, depth=0, cex.label = 2)