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）。