Search code examples
rxgboostmlr3

Element with key 'surv.xgboost' not found in DictionaryLearner


I am using R version 3.6.3, mlr3 version 0.3.0 and mlr3proba version 0.1.6 (the latest development versions I could find) and xgboost version 0.90.0.2 → I am trying to use the command: lrn("surv.xgboost") but I receive the error message:

Error: Element with key 'surv.xgboost' not found in DictionaryLearner! Did you mean 'classif.xgboost' / 'regr.xgboost'?

Please let me know if you would like me to provide any further details concerning the issue.


Solution

  • Thanks for your interest in mlr3proba :) The xgboost learner is currently in limbo between mlr3proba and mlr3learners, as of tomorrow you can simply run

    install.packages("mlr3learners")
    library(mlr3learners)
    lrn("surv.xgboost")
    

    but until then please instead run

    remotes::install_github("mlr-org/mlr3learners#135")
    library(mlr3learners)
    lrn("surv.xgboost")