Shape of the input of the Ordinal Loss

[mohuzi]

allRank/allrank/models/losses/ordinal.py

Line 28 in c90b64a

:param y_pred: predictions from the model, shape [batch_size, slate_length]

Shouldn't the shape be [batch_size, slate_length, n] in here?
Thanks,

[PrzemekPobrotyn]

Yes, I believe you're right! Thanks for spotting this, will fix it soon.

[sadaharu-inugami]

Solved in a32a9bc. If that's all, you can close the issue now.