cmcmaster1/torchtabular: Package for training tabular models using the {torch} package.

This contains a base tabular transformer model. Depending on the hyperparameters chosen, the model could be idential to the SAINT architecture (https://arxiv.org/abs/2106.01342), the TabTransformer architecture (https://arxiv.org/abs/2012.06678) or a hybrid.

Getting started

Package details

AuthorChristopher McMaster
MaintainerThe package maintainer <christopher.mcmaster@austin.org.au>
LicenseGPL (>= 3)
Version0.1.0
URL https://github.com/cmcmaster1/torchtabular
Package repositoryView on GitHub
Installation Install the latest version of this package by entering the following in R:
install.packages("remotes")
remotes::install_github("cmcmaster1/torchtabular")
cmcmaster1/torchtabular documentation built on Dec. 19, 2021, 5:17 p.m.