radi-cho / GatedTabTransformer-Applications

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GatedTabTransformer applied in practice

Applications for the GatedTabTransformer deep learning architecture developed initially for InnoFair 2022.

@misc{cholakov2022gatedtabtransformer,
      title={The GatedTabTransformer. An enhanced deep learning architecture for tabular modeling}, 
      author={Radostin Cholakov and Todor Kolev},
      year={2022},
      eprint={2201.00199},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

Fetal health

Data

Dataset used: https://www.kaggle.com/andrewmvd/fetal-health-classification

Results

Energy prices

Data

Dataset used: https://www.kaggle.com/nicholasjhana/energy-consumption-generation-prices-and-weather

Screen.Recording.2022-02-02.at.14.37.18.mov

Results

About


Languages

Language:Jupyter Notebook 100.0%