handy-nn 0.0.1

Delightful and useful neural networks models, including OrdinalRegressionLoss, etc.

handy-nn

Delightful and useful neural networks models, including OrdinalRegressionLoss, etc.

Install

$ pip install handy-nn

Usage

from handy_nn import OrdinalRegressionLoss

# Initialize the loss function
num_classes = 5
criterion = OrdinalRegressionLoss(num_classes)

# For training
logits = model(inputs)  # Shape: (batch_size, 1)
loss = criterion(logits, targets)
loss.backward()

# To get class probabilities
probas = criterion.predict_probas(logits)  # Shape: (batch_size, num_classes)

License

MIT