nhirons/deepordinal
Ordinal output layers and loss functions (Rennie & Srebro, 2005) for PyTorch, TF/Keras, and JAX/Flax.
GitHub repository with 14 stars and 2 forks.
Language: Python
Topics: deep-learning, flax, jax, keras, ordinal-regression, pytorch, tensorflow