metrics manipulation

IvanKuchin · Nov 25, 2024 · 08bdc41 · 08bdc41
1 parent c6484b6
commit 08bdc41
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/pancreas_ai/config.py b/src/pancreas_ai/config.py
@@ -20,7 +20,7 @@
 
 LOSS_FUNCTION = "dice" # "dice" or "scce"
 
-MONITOR_METRIC = "val_custom_f1"
+MONITOR_METRIC = "val_f1"
 MONITOR_MODE = "max"
 
 MODEL_CHECKPOINT = "checkpoints/weights.keras"

diff --git a/src/pancreas_ai/tools/craft_network/metrics.py b/src/pancreas_ai/tools/craft_network/metrics.py
@@ -13,7 +13,7 @@ def reset_states(self):
         self.m.assign(0)
 
     def update_state(self, y_true, y_pred, sample_weight=None):
-        print("y_true:", y_true.numpy(), "\t y_pred:", y_pred)
+        print("y_true:", y_true, "\t y_pred:", y_pred)
 
     def result(self):
         return self.m
@@ -24,7 +24,10 @@ def __classification_metrics(config: dict) -> list:
         tf.keras.metrics.Precision(name="precision"),
         tf.keras.metrics.Recall(name="recall"),
         tf.keras.metrics.F1Score(name="f1"),
-        binary_metrics.Binary_MCC(name = "custom_mcc")
+        tf.keras.metrics.TruePositives(name="tp"),
+        tf.keras.metrics.TrueNegatives(name="tn"),
+        binary_metrics.Binary_MCC(name = "custom_mcc"),
+        # PrintMetricsInput(),
         # binary_metrics.precision,
         # binary_metrics.recall,
         # binary_metrics.f1,