MachineShop (version 2.0.0)

predict: Model Prediction

Description

Predict outcomes with a fitted model.

Usage

# S3 method for MLModelFit
predict(
  object,
  newdata = NULL,
  times = NULL,
  type = c("response", "prob"),
  cutoff = MachineShop::settings("cutoff"),
  dist = NULL,
  method = NULL,
  ...
)

Arguments

object

model fit result.

newdata

optional data frame with which to obtain predictions. If not specified, the training data will be used by default.

times

numeric vector of follow-up times at which to predict survival events/probabilities or NULL for predicted survival means.

type

specifies prediction on the original outcome scale ("response") or on a probability distribution scale ("prob").

cutoff

numeric (0, 1) threshold above which binary factor probabilities are classified as events and below which survival probabilities are classified.

dist

character string specifying distributional approximations to estimated survival curves. Possible values are "empirical", "exponential", "rayleigh", or "weibull"; with defaults of "empirical" for predicted survival events/probabilities and "weibull" for predicted survival means.

method

character string specifying the empirical method of estimating baseline survival curves for Cox proportional hazards-based models. Choices are "breslow", "efron" (default), or "fleming-harrington".

...

arguments passed to model-specific prediction functions.

See Also

confusion, performance, metrics

Examples

# NOT RUN {
## Survival response example
library(survival)
library(MASS)

gbm_fit <- fit(Surv(time, status != 2) ~ sex + age + year + thickness + ulcer,
               data = Melanoma, model = GBMModel)
predict(gbm_fit, newdata = Melanoma, times = 365 * c(2, 5, 10), type = "prob")

# }