julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Rename loss functions

Browse Source
This commit is contained in:
Jensun Ravichandran 2021-06-04 22:15:57 +02:00
parent 827958a28a
commit 7bc0bfa3ab
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/functions/losses.py
View File

@ -74,7 +74,8 @@ def _get_class_probabilities(probabilities, targets, prototype_labels):
return whole, correct, wrong
def log_likelihood_ratio_loss(probabilities, targets, prototype_labels):
def nllr_loss(probabilities, targets, prototype_labels):
"""Compute the Negative Log-Likelihood Ratio loss."""
_, correct, wrong = _get_class_probabilities(probabilities, targets,
prototype_labels)
@ -83,7 +84,8 @@ def log_likelihood_ratio_loss(probabilities, targets, prototype_labels):
return -1.0 * log_likelihood
def robust_soft_loss(probabilities, targets, prototype_labels):
def rslvq_loss(probabilities, targets, prototype_labels):
"""Compute the Robust Soft Learning Vector Quantization (RSLVQ) loss."""
whole, correct, _ = _get_class_probabilities(probabilities, targets,
prototype_labels)