predict {MachineShop} | R Documentation |
Predict outcomes with a fitted model.
## S3 method for class 'MLModelFit' predict( object, newdata = NULL, times = NULL, type = c("response", "prob"), cutoff = MachineShop::settings("cutoff"), dist = NULL, method = NULL, ... )
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 |
type |
specifies prediction on the original outcome scale
( |
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 |
method |
character string specifying the empirical method of estimating
baseline survival curves for Cox proportional hazards-based models.
Choices are |
... |
arguments passed to model-specific prediction functions. |
confusion
, performance
,
metrics
## Requires prior installation of suggested package gbm to run ## Survival response example library(survival) gbm_fit <- fit(Surv(time, status) ~ ., data = veteran, model = GBMModel) predict(gbm_fit, newdata = veteran, times = c(90, 180, 360), type = "prob")