predict.mlr3superlearner: Predict method for 'mlr3superlearner' object

In mlr3superlearner: Super Learner Fitting and Prediction

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)
}

mlr3superlearner documentation built on Sept. 17, 2024, 5:08 p.m.