search

shiba24 / learning2rank

Learning to rank with neuralnet - RankNet and ListNet
481 stars 141 forks source link

input/output #8

Closed nickedes closed 6 years ago

nickedes commented 6 years ago

Hi Please expalin the input and output formats required for this, is the output of ranknet a probability or a rank? please clarify

shiba24 commented 6 years ago

Hi @nickedes , as you can see in the README,

Here, X is numpy array with the shape of (num_samples, num_features) and y is numpy array with the shape of (num_samples, ). y is the score which you would like to rank based on (e.g., Sales of the products, page view, etc).

Thanks!