mlr-org / mlr3extralearners

Extra learners for use in mlr3.
https://mlr3extralearners.mlr-org.com/
81 stars 48 forks source link

Add argument to `surv.xgboost` to differentiate between Cox and AFT objectives #316

Closed bblodfon closed 2 months ago

bblodfon commented 8 months ago
sebffischer commented 8 months ago

You can still add two different versions to the dictionary. mlr_learners$add("id1", function() LearnerSurvXGBoost$new(type = 1))

mlr_learners$add("id2", function() LearnerSurvXGBoost$new(type = 2))

otherwise the Dictionary -> data.table conversion might show slightly imprecise information on the learner regarding predict types :)