CODEWITHSUNDEEP
mlr3
Nip
Nip

Reputation: 420

How to get mlr3 importance scores from learner?

How do I get importance scores? I tried this:

task = tsk("iris")
search_space = ParamSet$new(
  params = list(
  ParamDbl$new(id = "eta", lower = 0.2, upper = .4)))

at = AutoTuner$new(
  learner = lrn("classif.xgboost"),
  resampling = rsmp("holdout"),
  measure = msr("classif.ce"),
  terminator = trm("evals", n_evals = 2),
  tuner = tnr("grid_search"),
  search_space = search_space,
  store_tuning_instance = TRUE)

at$train(task)
at$importance
at$importance()

but it seems not to work.

Upvotes: 0

Views: 338

Answers (1)

Lars Kotthoff
Lars Kotthoff

Reputation: 109242

You need to get the importance from the learner, not the tuner:

> at$learner$importance()
 Petal.Width Petal.Length 
   0.5028937    0.4971063

Upvotes: 2

Related Questions