yandex-research / rtdl-num-embeddings

(NeurIPS 2022) On Embeddings for Numerical Features in Tabular Deep Learning
https://arxiv.org/abs/2203.05556
MIT License
312 stars 34 forks source link

training config file #12

Closed morgsmss7 closed 1 year ago

morgsmss7 commented 1 year ago

Could you provide a sample 0.toml file? I am attempting to run the training scripts on new data and am hitting a few issues I did not hit in tune.py.

Yura52 commented 1 year ago

Do I understand correctly that you are trying to directly use bin/train4.py without bin/tune.py and you are looking for some kind of "default" hyperparameters? In particular, there are many algorithms in this repository (and in the paper), which one exactly you are trying to run?

morgsmss7 commented 1 year ago

Hello Yura,

A default train4.py toml file would be very helpful. Right now, I am using mlp and mlp-plrlr, so an example for those would be quite useful. We are essentially looking to repeat a few of your experiments on our datasets.

Best, Morgan

On May 16, 2023, at 6:10 AM, Yura Gorishniy @.***> wrote:

Do I understand correctly that you are trying to directly use bin/train4.py without bin/tune.py and you are looking for some kind of "default" hyperparameters? In particular, there are many algorithms in this repository (and in the paper), which one exactly you are trying to run?

— Reply to this email directly, view it on GitHub https://github.com/Yura52/tabular-dl-num-embeddings/issues/12#issuecomment-1549379732, or unsubscribe https://github.com/notifications/unsubscribe-auth/AJQFE4TYIG7YC37CG75SR2TXGNHAHANCNFSM6AAAAAAYCWN74A. You are receiving this because you authored the thread.

Yura52 commented 1 year ago

I see. Right now, I don't have an opportunity to build a fully working example, but you can notify me on May 18 if the help will still be needed.

Some quick comments:

P.S. If MLP-PLRLR will be too slow, you can try MLP-PLR

Yura52 commented 1 year ago

Feel free to reopen the issue if needed!