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

[WIP] Reimplement lambdamart ndcg. #6465

Closed
wants to merge 5 commits into from
Closed

Conversation

trivialfis
Copy link
Member

@trivialfis trivialfis commented Dec 4, 2020

Related: #6450

  • Add parameter ndcg_truncation.
  • Change the default metric to ndcg@truncation.

TODOS:

  • Better tests.
  • Benchmark for accuracy and perf.
  • Label checks.
  • Gain type.

@trivialfis
Copy link
Member Author

Question to myself: Is there a way to use cub sort pair with discard iterator to avoid allocating an array?

python-package/xgboost/dask.py Outdated Show resolved Hide resolved
* Simplify the implementation for both CPU and GPU.

Fix JSON IO.

Check labels.

Put idx into cache.

Optimize.

File tag.

Weights.

Trivial tests.

Compatibility.

Lint.

Fix swap.

Device weight.

tidy.

Easier to read R failure.

enum.

Fix global configuration.

Tidy.

msvc omp.

dask.

Remove ndcg specific parameter.  Drop label type for smaller PR.

Fix rebase.

Fixes.

Don't mess with includes.

Fixes.

Format.

Use omp util.

Restore some old code.
@trivialfis
Copy link
Member Author

Closing to avoid spamming the list. Will reopen once ready.

@trivialfis trivialfis closed this Jan 13, 2023
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.

2 participants