AutoCompBoostRegr: AutoCompBoostRegr

In Coorsaa/autocompboost: AutoML with compboost

AutoCompBoostRegr

Description

Class for Automated Regression in autocompboost. Subclass of AutoCompBoostBase

Creates a new instance of this R6 class.

Arguments

task	(Task) TaskRegr to be solved.
resampling	(Resampling) Contains the resampling method to be used for hyper-parameter optimization. Defaults to ResamplingCV with 3 folds.
param_values	(list()) Parameter values which are pass on to the learner.
measure	(Measure) Contains the performance measure, for which we optimize during training. Defaults to Accuracy for classification and RMSE for regression.
tuning_method	(character(1)) Tuning method. Possible choices are "mbo", "hyperband" or "smashy"¸ Default is "smashy".
tuning_time	(integer(1)) Termination criterium. Number of seconds for which to run the optimization. Does not include training time of the final model. Default is set to 3600, i.e. one hour. Tuning is terminated depending on the first termination criteria fulfilled.
tuning_iters	(integer(1)) Termination criterium. Number of MBO iterations for which to run the optimization. Default is set to 150 iterations. Tuning is terminated depending on the first termination criteria fulfilled.
tuning_generations	(integer(1)) Termination criterium for tuning method smashy. Number of generations for which to run the optimization. Default is set to 3 generations. Tuning is terminated depending on the first termination criteria fulfilled.
enable_tuning	(logical(1)) Whether or not to perform hyperparameter optimization. Default is TRUE.
final_model	(logical(1)) Whether or not to return the final model trained on the whole dataset at the end.

Value

AutoCompBoostRegr

Construction

Objects should be created using the AutoCompBoost interface function.

regression_model = AutoCompBoost(regression_task, resampling, measure,
tuning_time, tuning_iters, final_model)

Examples

## Not run: 
library(mlr3)
library(autocompboost)

regression_model = AutoCompBoost(tsk("boston_housing"))
regression_model$train()

## End(Not run)

Coorsaa/autocompboost documentation built on March 19, 2023, 12:08 p.m.