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

chore: scikit compatible version #42

Merged
merged 4 commits into from
Dec 3, 2019
Merged

Conversation

Optimox
Copy link
Collaborator

@Optimox Optimox commented Nov 29, 2019

IMPORTANT: Please do not create a Pull Request without creating an issue first.

Any change needs to be discussed before proceeding. Failure to do so may result in the rejection of the pull request.

What kind of change does this PR introduce?
Make TabNet Scikit compatible!

Does this PR introduce a breaking change?

What needs to be documented once your changes are merged?

Closing issues

Put closes #XXXX in your comment to auto-close the issue that your PR fixes (if such).
closes #41

Training and validation dataloaders
-------
"""
raise NotImplementedError('users must define construct_loaders to use this base class')

def fit(self, X_train, y_train, X_valid=None, y_valid=None,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Shouldn't we call load_best_model at the end of this function? I'm not sure this is the best solution, but it can be misleading if the output model after the fit is completed is not the one that performs best on the validation set.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I agree added it!

lambda_sparse=1e-3,
clip_value=1, verbose=1,
lr=2e-2, optimizer_fn=torch.optim.Adam,
lr_params={}, scheduler=None, scheduler_fn=None,
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Do we still have scheduler_params?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

disappeared by mistake I'm on it

@eduardocarvp eduardocarvp merged commit 3b1c4d9 into develop Dec 3, 2019
@Hartorn Hartorn deleted the chore/41-scikit-compatible branch March 18, 2020 10:32
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

Successfully merging this pull request may close these issues.

Make TabNet Scikit Compatible
2 participants