• sbruch's avatar
    Implementation of XE_NDCG_MART for the ranking task (#2620) · 86530988
    sbruch authored
    * 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
    86530988
test_consistency.py 5.36 KB