#' @export
makeRLearner.surv.ranger = function() {
makeRLearnerSurv(
cl = "surv.ranger",
package = "ranger",
par.set = makeParamSet(
makeIntegerLearnerParam(id = "num.trees", lower = 1L, default = 500L),
# FIXME: Add default value when data dependent defaults are implemented: mtry=floor(sqrt(#independent vars))
makeIntegerLearnerParam(id = "mtry", lower = 1L),
makeIntegerLearnerParam(id = "min.node.size", lower = 1L, default = 3L),
makeLogicalLearnerParam(id = "replace", default = TRUE),
makeNumericLearnerParam(id = "sample.fraction", lower = 0L, upper = 1L),
makeNumericVectorLearnerParam(id = "split.select.weights", lower = 0, upper = 1),
makeUntypedLearnerParam(id = "always.split.variables"),
makeDiscreteLearnerParam("respect.unordered.factors", values = c("ignore", "order", "partition"), default = "ignore"),
makeDiscreteLearnerParam(id = "importance", values = c("none", "permutation"), default = "none", tunable = FALSE),
makeLogicalLearnerParam(id = "write.forest", default = TRUE, tunable = FALSE),
makeLogicalLearnerParam(id = "scale.permutation.importance", default = FALSE, requires = quote(importance == "permutation"), tunable = FALSE),
makeIntegerLearnerParam(id = "num.threads", lower = 1L, when = "both", tunable = FALSE),
makeLogicalLearnerParam(id = "save.memory", default = FALSE, tunable = FALSE),
makeLogicalLearnerParam(id = "verbose", default = TRUE, when = "both", tunable = FALSE),
makeIntegerLearnerParam(id = "seed", when = "both", tunable = FALSE),
makeDiscreteLearnerParam(id = "splitrule", values = c("logrank", "extratrees", "C", "maxstat"), default = "logrank"),
makeIntegerLearnerParam(id = "num.random.splits", lower = 1L, default = 1L, requires = quote(splitrule == "extratrees")),
makeNumericLearnerParam(id = "alpha", lower = 0L, upper = 1L, default = 0.5, requires = quote(splitrule == "maxstat")),
makeNumericLearnerParam(id = "minprop", lower = 0, upper = 0.5, default = 0.1, requires = quote(splitrule == "maxstat")),
makeLogicalLearnerParam(id = "keep.inbag", default = FALSE, tunable = FALSE)
),
par.vals = list(num.threads = 1L, verbose = FALSE, respect.unordered.factors = "order"),
properties = c("numerics", "factors", "ordered", "featimp", "weights"),
name = "Random Forests",
short.name = "ranger",
note = "By default, internal parallelization is switched off (`num.threads = 1`), `verbose` output is disabled, `respect.unordered.factors` is set to `order` for all splitrules. All settings are changeable.",
callees = "ranger"
)
}
#' @export
trainLearner.surv.ranger = function(.learner, .task, .subset, .weights = NULL, ...) {
tn = getTaskTargetNames(.task)
ranger::ranger(formula = NULL, dependent.variable.name = tn[1L],
status.variable.name = tn[2L], data = getTaskData(.task, .subset), case.weights = .weights, ...)
}
#' @export
predictLearner.surv.ranger = function(.learner, .model, .newdata, ...) {
p = predict(object = .model$learner.model, data = .newdata)
rowMeans(p$chf)
}
#' @export
getFeatureImportanceLearner.surv.ranger = function(.learner, .model, ...) {
getFeatureImportanceLearner.classif.ranger(.learner, .model, ...)
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.