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.