Skip to content
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.

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

A scikit-learn interface for RTDL package. #30

Closed
hengzhe-zhang opened this issue Mar 31, 2022 · 2 comments
Closed

A scikit-learn interface for RTDL package. #30

hengzhe-zhang opened this issue Mar 31, 2022 · 2 comments

Comments

@hengzhe-zhang
Copy link

Hello! I have written a scikit-learn interface for the RTDL package (https://github.com/hengzhe-zhang/scikit-rtdl). I rely on the skorch to avoid coding errors, and set the default parameters based on the parameters presented in your paper. Hoping you will like it!

@Yura52
Copy link
Collaborator

Yura52 commented Apr 21, 2022

Hello! Sorry for the slow reply, and thanks for your work! I like that skorch is used. I should add that hyperparameters for specific models should be tuned, if possible. While we have the default configuration for FT-Transformer, we do not have it for other models (though, good defaults for them may exist as well!).

@Yura52
Copy link
Collaborator

Yura52 commented Dec 21, 2023

FYI, the rtdl package is now replaced with two packages:

The new packages only implement the papers, so scikit-learn interface is not planned for them. I have been keeping this issue open for quite some time to make it more visible, I hope it helped. Thank you for your repository!

@Yura52 Yura52 closed this as completed Dec 21, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants