Fix bug: MaxHingeTarget was inheriting loss='mse' from parent Target

PPPLDeepLearning · Jan 9, 2020 · eb8f664 · eb8f664
1 parent 8a36a58
commit eb8f664
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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):