kazk1018 / tf-listnet

learning to rank

tf-listnet

This repository is for my blog.

Requirements

• python 3.7
• pipenv

Dataset

LETOR: Learning to Rank for Information Retrieval (Microsoft)

• LETOR4.0/MQ2008
• supervised ranking
• Download from link
• Dataset should be downloaded in data/ at project root directory by default

tf-listnet/
  data/
    MQ2008/
      Fold1/
      Fold2/
      ...
      S1.txt
      S2.txt
      ...

Run

By default (only training model)

pipenv run datasets
pipenv run train

Training with evaluation on validation data

pipenv run datasets
pipenv run datasets --input data/MQ2008/Fold1/vali.txt --output data/validation.tfrecord
pipenv run train --validation data/validation.tfrecord