Model selection from a candidate set.
SelectedModel(
...,
control = MachineShop::settings("control"),
metrics = NULL,
stat = MachineShop::settings("stat.train"),
cutoff = MachineShop::settings("cutoff")
)
model functions, function names, calls, or vectors
of these to serve as the candidate set from which to select, such as that
returned by expand_model
.
control function, function name, or call defining the resampling method to be employed.
metric function, function name, or vector of these with which to calculate performance. If not specified, default metrics defined in the performance functions are used. Model selection is based on the first calculated metric.
function or character string naming a function to compute a summary statistic on resampled metric values for model selection.
argument passed to the metrics
functions.
SelectedModel
class object that inherits from MLModel
.
factor
, numeric
, ordered
,
Surv
# NOT RUN {
model_fit <- fit(sale_amount ~ ., data = ICHomes,
model = SelectedModel(GBMModel, GLMNetModel, SVMRadialModel))
(selected_model <- as.MLModel(model_fit))
summary(selected_model)
# }
Run the code above in your browser using DataLab