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

[DRAFT] added agumentation yielding evenly distributed classes #768

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

Conversation

TiMauzi
Copy link

@TiMauzi TiMauzi commented Dec 18, 2023

What does this PR do?

Summary

This PR adds a new augmentation method called augment_text_with_ids_evenly which yields a similar output to augment_text_with_ids, but always returns equal numbers of strings per ID. The argument additional_examples can increase the number of augmented samples further beyond the maximum number of examples over all classes. The argument perfectly_even asserts an exactly equal number of samples per class, but the augmentation will take slightly longer.

Additions

  • Added augment_text_with_ids_evenly.

Checklist

  • The title of your pull request should be a summary of its contribution.
  • Please write detailed description of what parts have been newly added and what parts have been modified. Please also explain why certain changes were made.
  • If your pull request addresses an issue, please mention the issue number in the pull request description to make sure they are linked (and people consulting the issue know you are working on it)
  • To indicate a work in progress please mark it as a draft on Github.
  • Make sure existing tests pass.
  • Add relevant tests. No quality testing = no merge.
  • All public methods must have informative docstrings that work nicely with sphinx. For new modules/files, please add/modify the appropriate .rst file in TextAttack/docs/apidoc.'

@TiMauzi
Copy link
Author

TiMauzi commented Dec 18, 2023

Please excuse the poor commit message, since it used to be "added augment_text_with_ids_evenly".

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.

1 participant