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

Add instructions to use a custom model #26

Open
Jesse-Islam opened this issue Oct 4, 2022 · 1 comment
Open

Add instructions to use a custom model #26

Jesse-Islam opened this issue Oct 4, 2022 · 1 comment

Comments

@Jesse-Islam
Copy link

I’m currently testing LassoNet as a potential model in my research and would like to use a custom model. Is there a recommended way to do so? I was scanning the examples and didn’t quite see one.

I am performing a [0,1] classification task on high dimensional data, and would like to use a custom data generator / specific activation functions / add layers.

@louisabraham
Copy link
Collaborator

I guess you would have to use the model in model.py and reproduce the training loop from interfaces.py.

We should probably add examples to do that!

@louisabraham louisabraham changed the title LassoNet question: Custom model Add instructions to use a custom model Nov 28, 2022
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