src.models.pom_head

Module for the custom implementation of Proportional Odds Model (POM) components.

Functions

pom_loss(cumulative_probs, y, num_classes)	Calculates the loss for the Proportional Odds Model (POM).
pom_predict(cumulative_probs)	Predicts class labels from cumulative probabilities for POM.

Classes

POMHead(input_size, num_classes)

A neural network head for the Proportional Odds Model (POM).