src.models.adjacent_head

Functions

adjacent_loss(probs, y)	Calculates the adjacent loss for ordinal regression.
adjacent_predict(probs)	Predicts class labels from probabilities.

Classes

AdjacentHead(input_size, num_classes)

A neural network head for adjacent category ordinal regression.