/tabnet-lightning

A high-level API extension of TabNet for training with PyTorch Dataloaders directly

Primary LanguagePythonMIT LicenseMIT

tabnet-lightning

A high-level API extension of TabNet for training with PyTorch Dataloaders directly