pyg-team / pytorch-frame

Tabular Deep Learning Library for PyTorch
https://pytorch-frame.readthedocs.io
MIT License
540 stars 54 forks source link

Initial Model Implementation & Reproduction #27

Closed weihua916 closed 1 year ago

weihua916 commented 1 year ago

We aim to include four popular models in the initial release. This issue keeps track of our progress.

FT-Transformer @kaidic @weihua916

Trompt @weihua916

TabNet @zechengz @weihua916

ExcelFormer @yiweny

TabTransformer @yiweny

Example encoding + self-attention + sum pooling @weihua916

weihua916 commented 1 year ago

Completed!