Package website: release | dev
mlr3automl is the Automated Machine Learning (AutoML) package of the
mlr3 ecosystem. It automatically selects the
most suitable machine learning algorithm and tunes its hyperparameters
for a given task. The package includes 10 learners from the
mlr3learners package, ranging from simple models like glmnet to more
powerful algorithms such as ranger and xgboost. Leveraging the
mlr3pipelines package, it constructs sophisticated preprocessing
graphs with multiple parallel branches, which are jointly optimized
using the mlr3tuning package. The optimization is driven by
Asynchronous Decentralized Bayesian Optimization (ADBO), enabling
efficient and scalable AutoML.
Install the development version from GitHub:
remotes::install_github("mlr-org/mlr3automl")library("mlr3automl")
rush_plan(n_workers = 2)
task = tsk("spam")
learner = lrn("classif.auto",
terminator = trm("evals", n_evals = 100)
)
learner$train(task)