Question

我正在从mlr包包装学习器classif.rpart生成的决策树中创建规则，如何像在rpart.rules中那样打印规则，以及如何像在rpart.plot中那样可视化

使用classif.rpart创建学习者，训练并拟合模型，尝试使用rpart.plot进行绘制并将错误说成不是rpart对象

dt_mod <- mlr::train(fused_dt, classif.task)
dt_mod
library(rpart.plot)
rpart.plot(dt_mod$learner.model, roundint=FALSE, varlen=3, type = 3, clip.right.labs = FALSE, yesno = 2)
rpart.rules(dt_mod$learner.model, roundint = FALSE)

除了要列出的规则和要可视化的树之外，我

Answer 1

您可以使用getLearnerModel()直接访问学习者诱导的模型：

iris.model = train(makeLearner("classif.rpart"), iris.task)
rpart.plot(getLearnerModel(iris.model))

如果您的学习者被吸引，则通过more.unwrap = TRUE选项：

rpart.plot(getLearnerModel(iris.model, more.unwrap = TRUE))

如何从mlr包中可视化并打印classif.rpart模型的规则，例如在rpart.plot和rpart.rules中？

1 个答案: