Also, the output prediction should include the rankings, not only the scores
Also, the output prediction should include the rankings, not only the scores