New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

TabR #130

Open

EpicUsaMan opened this issue Sep 28, 2024 · 1 comment

Assignees

Labels

Milestone

EpicUsaMan commented Sep 28, 2024

Include TabR architecture

EpicUsaMan added the enhancement label

AnFreTh self-assigned this

Collaborator

AnFreTh commented Sep 28, 2024

Orientate on implementation for TabR here: https://github.com/yandex-research/tabular-dl-tabr/blob/main/bin/tabr.py

Adjust training loop, to have forward pass take (x_num, x_cat, targets) as input.

AnFreTh added this to the v1.1.1 Add new models milestone

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment