Skip to contents

Make predictions from a stackgbm model object

Usage

# S3 method for stackgbm
predict(object, newx, threshold = 0.5, classes = c(1L, 0L), ...)

Arguments

object

stackgbm model object

newx

New predictor matrix

threshold

Decision threshold. Default is 0.5.

classes

The class encoding vector of the predicted outcome. The naming and order will be respected.

...

unused

Value

A list of two vectors presenting the predicted classification probabilities and predicted response.

Examples

# check the vignette for code examples