Skip to content

Instantly share code, notes, and snippets.

@RaphaelS1

RaphaelS1/search_net_autotuner.R

Last active April 18, 2021 08:13
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save RaphaelS1/08d4d15bb980327302bab24656e2bfa8 to your computer and use it in GitHub Desktop.
Save RaphaelS1/08d4d15bb980327302bab24656e2bfa8 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
search_net_autotuner.R
library(mlr3tuning)
create_autotuner <- function(learner) {
 AutoTuner$new(
  learner = learner,
  search_space = search_space,
  resampling = rsmp("holdout"),
  measure = msr("surv.cindex"),
  terminator = trm("evals", n_evals = 2),
  tuner = tnr("random_search")
)
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment