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

Support sklearn cross validation for ranker. #8859

Merged
merged 8 commits into from
Mar 6, 2023
Merged

Support sklearn cross validation for ranker. #8859

merged 8 commits into from
Mar 6, 2023

Commits on Mar 6, 2023

  1. Support sklearn cross validation for ranker. 

    - Add a convention for X to include a special `qid` column.

sklearn utilities consider only `X`, `y` and `sample_weight` for supervised learning
algorithms, but we need an additional qid array for ranking.

It's important to be able to support the cross validation function in sklearn since all
other tuning functions like grid search are based on cross validation.
    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    edd8b94 View commit details
    Browse the repository at this point in the history

  2. consistent document.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    593cecb View commit details
    Browse the repository at this point in the history

  3. Update.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    2905cd1 View commit details
    Browse the repository at this point in the history

  4. Support custom metric as well.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    f2f55eb View commit details
    Browse the repository at this point in the history

  5. lint.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    b335e13 View commit details
    Browse the repository at this point in the history

  6. test cudf.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    57215fc View commit details
    Browse the repository at this point in the history

  7. pylint.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    ad78621 View commit details
    Browse the repository at this point in the history

  8. duplicated check.

    @trivialfis
    trivialfis committed Mar 6, 2023
    Configuration menu
    Copy the full SHA
    44e9dd5 View commit details
    Browse the repository at this point in the history