fix macro-adjust

ChEB-AI · Dec 14, 2023 · 5df0060 · 5df0060
1 parent e90d66b
commit 5df0060
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/chebai/callbacks/epoch_metrics.py b/chebai/callbacks/epoch_metrics.py
@@ -72,7 +72,7 @@ def on_train_epoch_end(
             self.train_macro_adjust = self._calculate_macro_adjust(self.train_labels)
             if self.train_macro_adjust != 1:
                 print(
-                    f"some classes are missing in train set, calculating macro-scores with adjustment factor {macro_adjust}"
+                    f"some classes are missing in train set, calculating macro-scores with adjustment factor {self.train_macro_adjust}"
                 )
 
         pl_module.log(
@@ -119,7 +119,7 @@ def on_validation_epoch_end(
             self.val_macro_adjust = self._calculate_macro_adjust(self.val_labels)
             if self.val_macro_adjust != 1:
                 print(
-                    f"some classes are missing in val set, calculating macro-scores with adjustment factor {macro_adjust}"
+                    f"some classes are missing in val set, calculating macro-scores with adjustment factor {self.val_macro_adjust}"
                 )
 
         pl_module.log(

diff --git a/chebai/result/classification.py b/chebai/result/classification.py
@@ -137,7 +137,7 @@ def print_metrics(preds, labels, device, classes=None, top_k=10, markdown_output
     macro_adjust = total_classes / classes_present
     if classes_present != total_classes:
         print(
-            f"{total_classes - classes_present} are missing, calculating macro-scores with adjustment factor {macro_adjust}"
+            f"{total_classes - classes_present} classes are missing, calculating macro-scores with adjustment factor {macro_adjust}"
         )
     print(
         f"Macro-F1 on test set with {preds.shape[1]} classes: {f1_macro(preds, labels) * macro_adjust:3f}"