[ModelMetrics] Added AUC

YanSte · Sep 10, 2023 · 54385c9 · 54385c9
1 parent a9e271e
commit 54385c9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/skit/ModelMetrics.py b/src/skit/ModelMetrics.py
@@ -52,7 +52,7 @@ def val_metric_key(self):
         if self == Metric.ACCURACY:
             return "val_accuracy"
         elif self == Metric.AUC:
-            return "val_auc"
+            return "auc"
         elif self == Metric.VAL_AUC:
             return "val_auc"
         elif self == Metric.VAL_ACCURACY:
@@ -76,7 +76,7 @@ def plot_labels(self):
             }
         elif self == Metric.AUC or self == Metric.VAL_AUC:
             return {
-                'Accuracy': {
+                'AUC': {
                     'Training AUC': 'auc',
                     'Validation AUC': 'val_auc'
                 },