【发布时间】:2019-05-01 17:38:56
【问题描述】:
我正在使用 caret 包来训练我的模型。
我的模型工作正常。但是当我绘制决策树时,标签是空白的。我如何获得标签?
carMod <- train( FLAG ~.,data=df_train, method="rpart" )
plot(carMod$finalModel)
【问题讨论】:
标签: r decision-tree r-caret
我正在使用 caret 包来训练我的模型。
我的模型工作正常。但是当我绘制决策树时,标签是空白的。我如何获得标签?
carMod <- train( FLAG ~.,data=df_train, method="rpart" )
plot(carMod$finalModel)
【问题讨论】:
标签: r decision-tree r-caret
按照plot.rpart (https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/plot.rpart.html) 的帮助页面中的说明,使用以下代码:
plot(carMod$finalModel)
text(carMod$finalModel)
交替使用
library(rpart.plot)
rpart.plot(carMod$finalModel)
【讨论】:
你需要
plot(carMod$finalModel)
text(carMod$finalModel)
【讨论】:
?text.rpart。例如,use.n = TRUE 添加更多信息。至于根,在plot 调用中尝试类似margin = -2 的内容。请记住,您既没有提供预期的输出,也没有提供数据。另请参阅 rpart.plot、partykit 和 maptree 包。