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

Question

我正在根据 mlr package wrapper learner classif.rpart 生成的决策树创建规则，如何打印规则 rpart.rules 以及如何可视化 rpart.plot

使用 classif.rpart 创建了学习器，训练并拟合了模型，尝试使用 rpart.plot 绘图并说错误为 Not an rpart object

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？

How to visualize and print rules of the classif.rpart model from mlr package, like in rpart.plot and rpart.rules?

r

mlr