Implementation of XE_NDCG_MART for the ranking task (#2620)
* Implementation of XE_NDCG loss function for ranking. * Add citation * Check in example usage for xe_ndcg loss. * Seed the generator when a seed is provided in the config. Add unit-tests for xe_ndcg * Update documentation * Fix indentation * Address issues raised by reviewers. * Clean up include statements. * Fix issues raised by reviewers. * Regenerate parameters.rst * Add a note to explain that reproducing xe_ndcg results requires num_threads to be one. * Introduce objective_seed and use that in rank_xendcg instead of directly using seed * Change default value of objective_seed
Showing
examples/xendcg/README.md
0 → 100644
examples/xendcg/predict.conf
0 → 100644
examples/xendcg/rank.test
0 → 100644
This diff is collapsed.
examples/xendcg/rank.train
0 → 100644
This diff is collapsed.
examples/xendcg/train.conf
0 → 100644
Please register or sign in to comment