Skip to content

Extension Packages

Michel Lang edited this page Feb 2, 2021 · 55 revisions

This page collects packages or projects which extend mlr3. Open an issue if you want your project listed here.

Overview

See this figure for an encompassing overview. The current state on GitHub and on CRAN is summarized on the CI-Status wiki page.

Released

Black-box optimization toolkit.

Post-hoc analysis (tests, plots) for benchmarks, i.e. comparison of the results of multiple learners on multiple tasks.

Extensions for cluster analysis.

Additional data sets (and tasks).

DataBackend for dbplyr. Allow tasks to transparently operate on various data storage systems, including SQL data bases, Apache Spark or Google's BigQuery.

Variable selection filters.

Variable selection wrappers like sequential forward/backward search, exhaustive search or genetic algorithms.

Hyperparameter tuning via hyperband.

Additional learners for regression and classification.

Many classification and regression performance measures, implemented as simple functions. Package comes with very few dependencies so it can easily be used in third-party packages.

Connector to OpenML.

Pipelines and DAGs for preprocessing and building complex workflows.

Extensions for supervised probabilistic learning (this includes survival analysis).

Extensions for resampling spatio-temporal tasks.

Hyperparameter tuning via random search, grid search, ...

Meta-package for installing and loading released core packages.

Visualization via ggplot2's autoplot() function.

In Development

Connector between mlr3 and batchtools

Extension to source and manage additional learners from the mlr3learners organization on GitHub.

Extensions for time-series forecasting.

Extensions for deep learning via keras.

Hyperparameter tuning via model-based optimization (a.k.a. Bayesian optimization).

Extensions for ordinal regression.

Planned

Open an issue in the respective project if you are interested in starting or supporting the development.

Extensions for functional data analysis.

Extensions for multilabel classification.