Я обучил и протестировал классификатор дерева решений с пакетом mlr3 в R:
pred_probability = learner_DT$train(task_train)$predict(task_test)
Как я могу получить важность переменной из этой модели? Я пробовал learner_DT$importance(), но результат был named numeric(0)
Любая помощь приветствуется.
Ваш код выглядит правильно (вам не нужно вызывать $predict(), чтобы понять важность, но это не повредит). В документации rpart.object указано:
variable.importance: именованный числовой вектор, указывающий важность каждой переменной. (Присутствует только в том случае, если есть какие-либо разбиения.) При печати с помощью «summary.rpart» они масштабируются, чтобы добавить к 100.
Так что, возможно, вы поместили дерево без разбиений (вы должны увидеть это в распечатанном выводе learner_DT$model).
