The table below shows all tabular deep learning learners that are added by mlr3torch: An overview of all learners can also be found on the mlr-org website.

library(data.table)
library(mlr3torch)
content = as.data.table(mlr3::mlr_learners)[, c("key", "label", "task_type", "feature_types", "packages")]
mlr3torch_ids = names(getFromNamespace("mlr3torch_learners", "mlr3torch"))
content = content[key %in% mlr3torch_ids]
is_vision = sapply(content$packages, function(ps) "torchvision" %in% ps)
content$packages = lapply(content$packages, function(ps) setdiff(ps, c("mlr3torch", "torch", "mlr3")))
content_vision = content[is_vision, ]
content_rest = content[!is_vision, ]
knitr::kable(content_rest)


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.