/mlr3automl

Automated machine learning in mlr3

Primary LanguageRGNU Lesser General Public License v3.0LGPL-3.0

mlr3automl

mlr3automl is an AutoML package for R based on mlr3.
The first version is up and running, feedback is very welcome!

Watch our “UseR! 2021” presentation on Youtube for an introduction. The slides are here.

UseR! 2021 mlr3automl

Installation

Make sure to have the latest versions of the relevant mlr3 packages.

devtools::install_github('https://github.com/mlr-org/mlr3extralearners')
devtools::install_github('https://github.com/a-hanf/mlr3automl', dependencies = TRUE)

Using mlr3automl

You can create your AutoML learner by passing a classification or regression Task from mlr3.

iris_task <- tsk('iris')
model <- AutoML(iris_task)
model$train()

Documentation

The vignette is a good starting point.

For the function reference, check the roxygen documentation:

?mlr3automl::AutoML