mlr3
enables efficient, object-oriented programming on the building blocks of machine learning. It provides R6
objects for tasks, learners, resamplings, and measures. The package is geared towards scalability and larger datasets by supporting parallelization and out-of-memory data-backends like databases. While mlr3
focuses on the core computational operations, add-on packages provide additional functionality.
Extends the mlr3 package with a backend to transparently work with databases such as SQLite', DuckDB
', MySQL
', MariaDB
', or PostgreSQL
'. The package provides two additional backends: DataBackendDplyr
relies on the abstraction of package dbplyr to interact with most DBMS. DataBackendDuckDB
operates on DuckDB
data bases and also on Apache Parquet files.
This package provides an interface to OpenML.org
to list and download machine learning data, tasks and experiments. The OpenML
objects can be automatically converted to mlr3 objects. For a more sophisticated interface with more upload options, see the OpenML
package.
This package offers features plots for mlr3 objects such as tasks, learners, predictions, benchmark results, tuning instances and filters via the autoplot()
generic of ggplot2. The mlr3viz package draws plots with the viridis color palette and applies the minimal theme. Visualizations include barplots, boxplots, histograms, ROC curves, and precision-recall curves.
Extends the mlr3 ecosystem to functional analysis by adding support for irregular and regular functional data as defined in the tf package. The package provides PipeOps
for preprocessing functional columns and for extracting scalar features, thereby allowing standard machine learning algorithms to be applied afterwards. Available operations include simple functional features such as the mean or maximum, smoothing, interpolation, flattening, and functional PCA'.
This package provides a flexible approach to Bayesian optimization / model based optimization building on the bbotk package. The mlr3mbo is a toolbox providing both ready-to-use optimization algorithms as well as their fundamental building blocks allowing for straightforward implementation of custom algorithms. Single- and multi-objective optimization is supported as well as mixed continuous, categorical and conditional search spaces. Moreover, using mlr3mbo for hyperparameter optimization of machine learning models within the mlr3 ecosystem is straightforward via mlr3tuning.
mlr3misc
provides frequently used helper functions and assertions used in mlr3
and its companion packages. It comes with helper functions for functional programming, for printing, to work with data.table
, as well as some generally useful R6
classes. This package also supersedes the package BBmisc
.
This package provides a small collection of interesting and educational machine learning data sets which are used as examples in the mlr3 book Applied machine learning using mlr3 in R https://mlr3book.mlr-org.com, the use case gallery https://mlr3gallery.mlr-org.com, or in other examples. All data sets are properly preprocessed and ready to be analyzed by most machine learning algorithms. Data sets are automatically added to the dictionary of tasks if mlr3 is loaded.
This package provides a web-based graphical user interface to provide the basic steps of a machine learning workflow. It uses the functionalities of the mlr3 framework.
The mlr3 package family is a set of packages for machine-learning purposes built in a modular fashion. This wrapper package is aimed to simplify the installation and loading of the core mlr3 packages.
Deep Learning library that extends the mlr3 framework by building upon the torch package. It allows to conveniently build, train, and evaluate deep learning models without having to worry about low level details. Custom architectures can be created using the graph language defined in mlr3pipelines'.
mlr3tuning
implements methods for hyperparameter tuning, e.g. Grid Search, Random Search, or Simulated Annealing. Various termination criteria can be set and combined. The class AutoTuner
provides a convenient way to perform nested resampling in combination with mlr3
.
This package extends the mlr3 package with cluster analysis.
This package extends mlr3
with support for performing ordinal regression.
Extends the mlr3 ML framework with methods for spatial objects. Data storage and prediction are supported for packages terra', raster and stars'.
This package extends mlr3 with filter methods for feature selection. Besides standalone filter methods built-in methods of any machine-learning algorithm are supported. Partial scoring of multivariate filter methods is supported.
Concise and interpretable summaries for machine learning models and learners of the mlr3 ecosystem. The package takes inspiration from the summary function for (generalized) linear models but extends it to non-parametric machine learning models, based on generalization performance, model complexity, feature importances and effects, and fairness metrics.
This package is a feature selection package of the mlr3 ecosystem. It selects the optimal feature set for any mlr3 learner. The package works with several optimization algorithms e.g. random search, Recursive feature elimination, and genetic search. Moreover, it can automatically optimize learners and estimate the performance of optimized feature sets with nested resampling.
This package implements multiple performance measures for supervised learning. It includes over 40 measures for regression and classification. Additionally, meta information about the performance measures can be queried, e.g. what the best and worst possible performances scores are.
mlr3learners
extends mlr3
and mlr3proba
with interfaces to essential machine learning packages on CRAN. This includes, but is not limited to: (penalized) linear and logistic regression, linear and quadratic discriminant analysis, k-nearest neighbors, naive Bayes, support vector machines, and gradient boosting.
Integrates fairness auditing and bias mitigation methods for the mlr3 ecosystem. This includes fairness metrics, reporting tools, visualizations and bias mitigation techniques such as "Reweighing" described in Kamiran, Calders (2012) <doi:10.1007/s10115-011-0463-8> and "Equalized Odds" described in Hardt et al. (2016) <https://papers.nips.cc/paper/2016/file/9d2682367c3935defcb1f9e247a97c0d-Paper.pdf>. Integration with mlr3 allows for auditing of ML models as well as convenient joint tuning of machine learning algorithms and debiasing methods.
This package implements methods for post-hoc analysis and visualisation of benchmark experiments, for mlr3 and beyond.
Extends the mlr3 package with a connector to the package batchtools'. This allows to run large-scale benchmark experiments on scheduled high-performance computing clusters.
This package implements a successive halving and hyperband optimization algorithm for the mlr3 ecosystem. The implementation in mlr3hyperband features improved scheduling and parallelizes the evaluation of configurations. The package includes tuners for hyperparameter optimization in mlr3tuning and optimizers for black-box optimization in bbotk.