I have a created a classification model with a custom ML framework and have the following:
For simplicity I have 3 classes: