Skip to content

remove sampler for valid dataloader in pretrain - #498

Open
cyang31 wants to merge 1 commit into
dreamquark-ai:developfrom
cyang31:develop
Open

cyang31 wants to merge 1 commit into
dreamquark-ai:developfrom
cyang31:develop

Conversation

@cyang31

@cyang31 cyang31 commented Jul 12, 2023 •

Copy link
Copy Markdown

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

This branch has not been deployed

No deployments
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