mlr3torch: Deep Learning with 'mlr3' (original) (raw)

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'.

Version:

0.2.1

Depends:

mlr3 (≥ 0.20.0), mlr3pipelines (≥ 0.6.0), torch (≥ 0.14.0), R (≥ 3.5.0)

Imports:

backports, checkmate (≥ 2.2.0), data.table, lgr, methods, mlr3misc (≥ 0.14.0), paradox (≥ 1.0.0), R6, withr

Suggests:

callr, curl, future, ggplot2, igraph, jsonlite, knitr, mlr3tuning (≥ 1.0.0), progress, rmarkdown, rpart, viridis, visNetwork, testthat (≥ 3.0.0), tfevents, torchvision (≥ 0.6.0), waldo

Published:

2025-02-13

DOI:

10.32614/CRAN.package.mlr3torch

Author:

Sebastian Fischer ORCID iD [cre, aut], Bernd Bischl ORCID iD [ctb], Lukas Burk ORCID iD [ctb], Martin Binder [aut], Florian Pfisterer ORCID iD [ctb], Carson Zhang [ctb]

Maintainer:

Sebastian Fischer <sebf.fischer at gmail.com>

BugReports:

https://github.com/mlr-org/mlr3torch/issues

License:

LGPL (≥ 3)

Copyright:

see file

URL:

https://mlr3torch.mlr-org.com/,https://github.com/mlr-org/mlr3torch/

NeedsCompilation:

no

Materials:

README NEWS

In views:

MachineLearning

CRAN checks:

mlr3torch results