Calls gbm::gbm from package gbm.

Dictionary

This Learner can be instantiated via the dictionary mlr_learners or with the associated sugar function lrn():

mlr_learners$get("surv.gbm")
lrn("surv.gbm")

Traits

  • Packages: gbm

  • Predict Types: crank, lp

  • Feature Types: integer, numeric, factor, ordered

  • Properties: importance, missings, weights

Custom mlr3 defaults

  • distribution:

  • Actual default: "bernoulli"

  • Adjusted default: "coxph"

  • Reason for change: This is the only distribution available for survival.

  • keep_data:

    • Actual default: TRUE

    • Adjusted default: FALSE

    • Reason for change: keep_data = FALSE saves memory during model fitting.

  • n.cores:

    • Actual default: NULL

    • Adjusted default: 1

    • Reason for change: Suppressing the automatic internal parallelization if cv.folds > 0.

See also

Author

RaphaelS1

Super classes

mlr3::Learner -> mlr3proba::LearnerSurv -> LearnerSurvGBM

Methods

Public methods

Inherited methods

Method new()

Creates a new instance of this R6 class.

Usage

LearnerSurvGBM$new()


Method importance()

The importance scores are extracted from the model slot variable.importance.

Usage

LearnerSurvGBM$importance()

Returns

Named numeric().


Method clone()

The objects of this class are cloneable with this method.

Usage

LearnerSurvGBM$clone(deep = FALSE)

Arguments

deep

Whether to make a deep clone.

Examples

# stop example failing with warning if package not installed learner = suppressWarnings(mlr3::lrn("surv.gbm")) print(learner)
#> <LearnerSurvGBM:surv.gbm> #> * Model: - #> * Parameters: distribution=coxph, keep.data=FALSE, n.cores=1 #> * Packages: gbm #> * Predict Type: crank #> * Feature types: integer, numeric, factor, ordered #> * Properties: importance, missings, weights
# available parameters: learner$param_set$ids()
#> [1] "distribution" "n.trees" "cv.folds" #> [4] "interaction.depth" "n.minobsinnode" "shrinkage" #> [7] "bag.fraction" "train.fraction" "keep.data" #> [10] "verbose" "var.monotone" "n.cores" #> [13] "single.tree"