r - mlr3 rpart 学习器中未定义变量重要性

Question

我用 R 中的 mlr3 包训练和测试了一个决策树分类器：

pred_probability = learner_DT$train(task_train)$predict(task_test)

我怎样才能从这个模型中得到变量的重要性？我试过learner_DT$importance()但结果是named numeric(0)

任何帮助表示赞赏。

Answer 1

您的代码看起来正确（您无需调用$predict()即可了解重要性，但这并没有什么坏处）。rpart.object 的文档指出：

variable.importance：一个命名的数字向量，给出每个变量的重要性。（仅在有任何拆分时才存在。）当由“summary.rpart”打印时，这些被重新缩放以添加到 100。

因此，您可能已经安装了一棵没有分裂的树（您应该在 的打印输出中看到这一点learner_DT$model）。