XE_NDCG Ranking Example

Here is an example for LightGBM to train a ranking model with the XE_NDCG loss.

You must follow the installation instructions for the following commands to work. The lightgbm binary must be built and available at the root of this project.

Training

Run the following command in this folder:

"../../lightgbm" config=train.conf

Prediction

You should finish training first.

Run the following command in this folder:

"../../lightgbm" config=predict.conf

Data Format

To learn more about the query format used in this example, check out the query data format.