Skip to content
Marc Becker edited this page Mar 20, 2020 · 60 revisions

Additional learners for mlr3

The following learners are shipped in their own packages to keep the {mlr3learners} package small.
Not all of them have necessarily been developed by the mlr-org team.
Also, learners might possibly be maintained by external persons.

See the section in the {mlr3learners} README on how to add/request a new learner.

Fully implemented

CI mlr3 Package Package and Function(s) Maintainer
R CMD Check via {tic} mlr3learner.ctree partykit:ctree Marc Becker (@be-marc)
R CMD Check via {tic} mlr3learners.extratrees extraTrees::extraTrees Marc Becker (@be-marc)
R CMD Check via {tic} mlr3learners.fnn FNN::knn Marc Becker (@be-marc)
R CMD Check via {tic} mlr3learners.liblinear LiblineaR::LiblineaR Marc Becker (@be-marc)
R CMD Check via {tic} mlr3learners.mboost mboost::gamboost mboost::glmboost Marc Becker (@be-marc)

Awaits review

These learners are working but need to be reviewed. Note that you probably have to switch the branch to get the latest release.

CI mlr3 Package Package and Function(s) Maintainer
mlr3learners.dbarts dbarts::bart Chris Kennedy (@ck37)
mlr3learners.earth earth::earth Philipp Kopper (@pkopper)
R CMD Check via {tic} mlr3learners.gbm gbm::gbm Marc Becker (@be-marc)
R CMD Check via {tic} mlr3learners.ksvm kernlab::ksvm Marc Becker (@be-marc)
mlr3learners.lightgbm lightgbm::lgb.train, lightgbm::lgb.cv Lorenz Kapsner (@kapsner)
mlr3learners.randomForest randomForest::randomForest Philipp Kopper (@pkopper)

In progress

CI mlr3 Package Package and Function(s) Maintainer
mlr3learners.h2o h2o::h2o.glm, h2o::h2o.glrm, h2o::h2o.gbm, h2o::h2o.randomForest, h2o::h2o.deeplearning Marc Becker (@be-marc)
mlr3learners.countglm stats::glm Philipp Kopper (@pkopper)
mlr3learners.Rborist Rborist::Rborist Philipp Kopper (@pkopper)

Installation

It is possible to have a CRAN-like installation via install.packages() by installing the learners from mlr3learners.drat.

Wishlist

  • classif.C50
  • cforest, should be with ctree in one "partykit" package
  • classif.JRip
  • classif.nnet
  • all algos from klaR (that mlr2 had)
  • regr.cubits
  • regr.earth
  • every MODERN COOL gaussian process you can find in R

Clone this wiki locally