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.

Sign up for GitHub

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

Jump to bottom

Added stratify option to train_test_split function. #4322

Merged
merged 16 commits into from
May 25, 2022
Merged

Added stratify option to train_test_split function. #4322

merged 16 commits into from
May 25, 2022

Commits on May 17, 2022

  1. Add stratify option to train_test_split

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    70da9b4 View commit details
    Browse the repository at this point in the history

  2. Add utility functions for performing stratified split

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    375eb9a View commit details
    Browse the repository at this point in the history

  3. Removed unused import

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    a831336 View commit details
    Browse the repository at this point in the history

  4. Add suggested changes

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    ab9522c View commit details
    Browse the repository at this point in the history

  5. Remove unused import from splits.py

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    b4ef6ad View commit details
    Browse the repository at this point in the history

  6. Add example usage of train_test_split with stratify arg to docstring

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    71bcbec View commit details
    Browse the repository at this point in the history

  7. Add test cases to test stratified_train_test_split

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    eb04b58 View commit details
    Browse the repository at this point in the history

  8. Move stratify functions to utils/stratify.py and refactor code.

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    049b51b View commit details
    Browse the repository at this point in the history

  9. Fix test cases according to ClassLabel class

    @nandwalritik
    nandwalritik committed May 17, 2022
    Configuration menu
    Copy the full SHA
    974b6dd View commit details
    Browse the repository at this point in the history

Commits on May 19, 2022

  1. Add changes for error handling and recommended changes

    @nandwalritik
    nandwalritik committed May 19, 2022
    Configuration menu
    Copy the full SHA
    faec0bb View commit details
    Browse the repository at this point in the history

  2. Add error handling for KeyErr for stratify_by_column arg

    @nandwalritik
    nandwalritik committed May 19, 2022
    Configuration menu
    Copy the full SHA
    8a6a299 View commit details
    Browse the repository at this point in the history

  3. Add tests for checking error handling in stratified train_test_split

    @nandwalritik
    nandwalritik committed May 19, 2022
    Configuration menu
    Copy the full SHA
    b409214 View commit details
    Browse the repository at this point in the history

  4. Removed unwanted imports

    @nandwalritik
    nandwalritik committed May 19, 2022
    Configuration menu
    Copy the full SHA
    2fe77cf View commit details
    Browse the repository at this point in the history

Commits on May 20, 2022

  1. Remove import datasets

    @mariosasko
    mariosasko authored May 20, 2022
    Configuration menu
    Copy the full SHA
    33208a0 View commit details
    Browse the repository at this point in the history

Commits on May 25, 2022

  1. Merge branch 'master' into add_stratify_train_test_split

    @lhoestq
    lhoestq authored May 25, 2022
    Configuration menu
    Copy the full SHA
    c9ee5c0 View commit details
    Browse the repository at this point in the history

  2. Update src/datasets/arrow_dataset.py

    @lhoestq
    lhoestq authored May 25, 2022
    Configuration menu
    Copy the full SHA
    48fe48e View commit details
    Browse the repository at this point in the history