mlr-org / mlr3hyperband

Successive Halving and Hyperband in the mlr3 ecosystem
https://mlr3hyperband.mlr-org.com
GNU Lesser General Public License v3.0
18 stars 5 forks source link

Set eta parameter for hyperband inside autotuner #92

Closed MislavSag closed 2 years ago

MislavSag commented 2 years ago

How can I set eta parameter inside autoruner ?

at_ranger = auto_tuner(
  method = "hyperband",
  learner = graph_learner,
  resampling = rsmp("cv", folds = 3),
  measure = msr("classif.ce"),
  search_space = search_space
)

method accepts only character, but I would like to pass tuner tnr("hyperband", eta = 5)

be-marc commented 2 years ago
at_ranger = auto_tuner(
  method = "hyperband",
  learner = graph_learner,
  resampling = rsmp("cv", folds = 3),
  measure = msr("classif.ce"),
  search_space = search_space,
  eta = 3
)

Additional arguments ... are passed to the tuner.