src.models.coral_head

Functions

coral_loss(logits, levels)	Computes the CORAL loss.
coral_predict(logits)	Makes predictions from CORAL logits.

Classes

OrdinalHead(input_size, num_classes)

Head for CORAL and CORN which expect num_classes - 1 outputs.