Mlr3torch - Deep Learning in R

MCML Authors

Sebastian Fischer

→ Group Bernd Bischl
 Statistical Learning and Data Science

Martin Binder

→ Group Bernd Bischl
 Statistical Learning and Data Science

Abstract

mlr3torch is a deep learning framework for the mlr3 ecosystem built on top of torch. It allows to easily build, train and evaluate deep learning models in a few lines of codes, without needing to worry about low-level details. Off-the-shelf learners are readily available, but custom architectures can be defined by connecting PipeOpTorch operators in an mlr3pipelines::Graph.

inproceedings FB24

---

useR! 2024

International R User Conference. Salzburg, Austria, Jul 08-11, 2024.

---

Authors

S. F. Fischer • M. Binder

---

Links

GitHub

---

Research Area

 A1 | Statistical Foundations & Explainability