Extension Packages

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.

Released

bbotk

Black-box optimization toolkit.

mlr3cluster

Extensions for cluster analysis.

mlr3data

Additional data sets (and tasks).

mlr3db

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

mlr3filters

Variable selection filters.

mlr3fselect

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

mlr3hyperband

Hyperparameter tuning via hyperband.

mlr3learners

Additional learners for regression and classification.

mlr3measures

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.

mlr3oml

Connector to OpenML.

mlr3pipelines

Pipelines and DAGs for preprocessing and building complex workflows.

mlr3proba

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

mlr3tuning

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

mlr3verse

Meta-package for installing and loading released core packages.

mlr3viz

Visualization via ggplot2's autoplot() function.

In Development

mlr3batchmark

Connector between mlr3 and batchtools

mlr3forecasting

Extensions for time-series forecasting.

mlr3keras

Extensions for deep learning via keras.

mlr3mbo

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

mlr3ordinal

Extensions for ordinal regression.

mlr3spatiotempcv

Extensions for resampling spatio-temporal tasks.

Planned

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

mlr3extralearners

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

mlr3fda

Extensions for functional data analysis.

mlr3multilabel

Extensions for multilabel classification.