How to Perform Ordinal Regression / Classification in PyTorch

A simple trick for improving model performance when labels are ordered.

Mathias Gruber
Towards Data Science
7 min readJul 10, 2021

--

Abstract vector created by vectorjuice — www.freepik.com

When you have a multiclass classification problem and there is an order to the classes, it is known as an “ordinal regression” problem. An example could be the classification of a student’s performance into categories A > B > C > D > E. The…

--

--