select_models: Select models from train() output.

In ModelOriented/forester: Quick and Simple Tools for Training and Testing of Tree-Based Models

Select models from train() output.

Description

The function provides a necessary interface for minimizing the size of output created with train(). It not only selects the models, but also the data, predictions, and all other necessary information.

Usage

select_models(train_output, models)

Arguments

train_output	The output created with train() function.
models	The vector including either the names of models or their indexes, that we want to select.

Value

The train_output with selected models.

Examples

## Not run: 
out      <- train(iris, 'Species', random_evals = 30, bayes_iters = 0)
new_out  <- select_models(out, c('random_forest_RS_1', 'xgboost_RS_14'))
new_out2 <- select_models(out, c(1, 15, 30, 45))

## End(Not run)

ModelOriented/forester documentation built on June 6, 2024, 7:29 a.m.