Skip to content

a-hanf/mlr3automl

Repository files navigation

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

About

Automated machine learning in mlr3

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages