predict.mlr3superlearner {mlr3superlearner}R Documentation

Predict method for mlr3superlearner object

Description

Predict method for mlr3superlearner object

Usage

## S3 method for class 'mlr3superlearner'
predict(object, newdata, ...)

Arguments

object

[mlr3superlearner]
An object returned from mlr3superlearner().

newdata

data [data.frame]
A data.frame containing predictors.

...

Unused.

Value

A vector of the predicted values.

See Also

mlr3superlearner

Examples

if (requireNamespace("ranger", quietly = TRUE)) {
  n <- 1e3
  W <- matrix(rnorm(n*3), ncol = 3)
  A <- rbinom(n, 1, 1 / (1 + exp(-(.2*W[,1] - .1*W[,2] + .4*W[,3]))))
  Y <- rbinom(n,1, plogis(A + 0.2*W[,1] + 0.1*W[,2] + 0.2*W[,3]^2 ))
  tmp <- data.frame(W, A, Y)
  fit <- mlr3superlearner(tmp, "Y", c("glm", "ranger"), "binomial")
  predict(fit, tmp)
}

[Package mlr3superlearner version 0.1.2 Index]