lucidrains / tab-transformer-pytorch

Implementation of TabTransformer, attention network for tabular data, in Pytorch
MIT License
808 stars 102 forks source link

Implement category specific embedding #13

Open LLYX opened 2 years ago

LLYX commented 2 years ago

Change each embedding to include a value specific component and a category specific component.