diff --git a/plasma/models/targets.py b/plasma/models/targets.py index fe1351e4..fc6729f9 100644 --- a/plasma/models/targets.py +++ b/plasma/models/targets.py @@ -126,6 +126,7 @@ def threshold_range(T_warning): # time sequence is punished. Also implements class weighting class MaxHingeTarget(Target): activation = 'linear' + loss = 'hinge' fac = 1.0 @staticmethod @@ -198,7 +199,7 @@ def threshold_range(T_warning): class HingeTarget(Target): activation = 'linear' - loss = 'hinge' # hinge + loss = 'hinge' @staticmethod def loss_np(y_true, y_pred):