Where are PLE embeddings?

yandex-research / rtdl-num-embeddings

(NeurIPS 2022) On Embeddings for Numerical Features in Tabular Deep Learning

https://arxiv.org/abs/2203.05556

MIT License

269 stars 31 forks source link

Where are PLE embeddings? #3

Closed Froskekongen closed 2 years ago

Froskekongen commented 2 years ago

I can't seem to find the piecewise linear embeddings (or utilities to create bins) in your repository(/ies). Are they here, but hidden, or do you plan to implement them later?

Yura52 commented 2 years ago

They are implemented in bin/train4.py, this comment provides more details. Feel free to reopen the issue or ask additional questions in that discussion.