allegro/allRank

Example to run listNet

sahilm1992 opened this issue · 3 comments

sahilm1992 commented

Is there any example on how to run the ListNet module?

sahilm1992 commented

Also is the code of Listnet correct?

sadaharu-inugami commented

The code of Listnet is correct, it is a vanilla implementation of the formula (2) in the paper Learning to Rank: From Pairwise Approach to Listwise Approach.

As for the example, you could modify the local config scripts/local_config.json and run the example script scripts/run_in_docker.sh, remembering about building the docker image first (docker build -t allrank:latest . from the main directory). The modification could limit just to changing the loss section from

  "loss": {
    "name": "ordinal",
    "args": {
    }
  }

to

  "loss": {
    "name": "listNet",
    "args": {
    }
  }
sahilm1992 commented