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

feat: add augmentations inside the fit method #363

Merged
merged 3 commits into from Mar 23, 2022

Conversation

Optimox
Copy link
Collaborator

@Optimox Optimox commented Mar 3, 2022

What kind of change does this PR introduce?
This adds the possibility to perform batch data augmentation during training.
SMOTE augmentation are proposed as examples.

Does this PR introduce a breaking change?
It's just a new possibility, no breaking change.

What needs to be documented once your changes are merged?
The README should probably be updated to document the new possibilities.

Closing issues

No issue related to this.

Makefile Show resolved Hide resolved
pytorch_tabnet/augmentations.py Show resolved Hide resolved
@Optimox Optimox changed the title WIP feat: add augmentations inside the fit method feat: add augmentations inside the fit method Mar 23, 2022
@eduardocarvp eduardocarvp merged commit 40107a8 into develop Mar 23, 2022
@eduardocarvp eduardocarvp deleted the feat/data-augmentation branch March 23, 2022 16:53
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.

None yet

2 participants