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

remove sampler for valid dataloader in pretrain #498

Open
wants to merge 1 commit into
base: develop
Choose a base branch
from

Conversation

cyang31
Copy link

@cyang31 cyang31 commented Jul 12, 2023

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.

The validation dataloader in pretrain step is not necessary. The current sampler in pretraining_utils.py line62 causes errors when X_train and X_valid in eval_sets don't have the same size. See #499

What kind of change does this PR introduce?
a bugfix

Does this PR introduce a breaking change?
No

What needs to be documented once your changes are merged?
Maybe add documents in the weights parameter of TabNetPretrainer.fit to say a 1d ndarray with the same length as the X_train.

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

@cyang31 cyang31 marked this pull request as ready for review July 12, 2023 21:55
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.

sampler is not needed in pretrain mode for valid dataloader
1 participant