train with ups dataset2

francesco-vaselli · Aug 28, 2023 · a95c55f · a95c55f
1 parent ff00e8f
commit a95c55f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils.py b/src/utils.py
@@ -164,7 +164,7 @@ def on_epoch_end(self, epoch, logs=None):
         y_true = self.test_y
 
         true_label, pred_label, fpr, tpr, roc_auc = check_classification(
-            y_true, y_pred, self.threshold, self.std, self.mean
+            y_true, y_pred, self.threshold, std=self.std, mean=self.mean
         )
 
         with self.writer.as_default():