radi-cho / GatedTabTransformer

A deep learning tabular classification architecture inspired by TabTransformer with integrated gated multilayer perceptron.
MIT License
95 stars 7 forks source link

Reuse utils, methods and classes across gMLP and TabTransformer implementations. #2

Closed radi-cho closed 2 years ago

radi-cho commented 2 years ago

Currently utility functions such as exists = lambda val : val is not None and block classes - PreNorm, Residual, etc. are present in both g_mlp.py, gated_tab_transformer.py.

radi-cho commented 2 years ago

https://github.com/radi-cho/GatedTabTransformer/commit/5dcaf0a0d6398bf9004c7441fe9367e72b9fa426