adapt metrics per class

ssciwr · Apr 9, 2024 · f7f110e · f7f110e
1 parent c362702
commit f7f110e
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/moralization/transformers_model_manager.py b/moralization/transformers_model_manager.py
@@ -19,7 +19,6 @@
 import frontmatter
 from huggingface_hub import HfApi
 import shutil
-from sklearn.metrics import classification_report
 
 
 IGNORED_LABEL = -100
@@ -319,9 +318,7 @@ def compute_metrics(self, eval_preds: tuple) -> Dict:
             "recall": all_metrics["overall_recall"],
             "f1": all_metrics["overall_f1"],
             "accuracy": all_metrics["overall_accuracy"],
-            "classification report": classification_report(
-                y_true=true_labels, y_pred=true_predictions
-            ),
+            "per class": all_metrics,
         }
 
     def _set_id2label(self) -> None: