Skip to content
This repository has been archived by the owner on Jun 23, 2023. It is now read-only.
/ xe-ndcg-experiments Public archive

Code to reproduce the experiments reported in "An Alternative Cross Entropy Loss for Learning-to-Rank" (https://arxiv.org/abs/1911.09798)

License

Notifications You must be signed in to change notification settings

sbruch/xe-ndcg-experiments

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Build LightGBM and set up Virtualenv

git clone --recursive https://github.com/microsoft/LightGBM
cd LightGBM
mkdir build ; cd build
cmake ..
make -j64

cd ../python-package

sudo apt-get install virtualenv python3-venv
python3 -m venv /tmp/xendcg
source /tmp/xendcg/bin/activate
pip install scipy scikit-learn numpy wheel
python setup.py install

Running Experiments

You may reproduce:

  • quality experiments:

    python main.py --paths /path/to/data/*libsvm
  • experiments with 20% of labels perturbed:

    python main.py --label_noise 0.2 --paths /path/to/data/*libsvm
  • experiments where each query is augmented by 40% of its original list size:

    python main.py --negs_from_others 0.4 --paths /path/to/data/*libsvm
  • experiments where 10 impressions are generated for each query with click probabilities conditioned on labels [.05, .3, .5, .7, .95] (i.e., documents with label 0 are clicked with 5% probability):

    python main.py --click_impressions 10 --click_probs .05 .3 .5 .7 .95 --paths /path/to/data/*libsvm

About

Code to reproduce the experiments reported in "An Alternative Cross Entropy Loss for Learning-to-Rank" (https://arxiv.org/abs/1911.09798)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages