Merge pull request #149 from ssciwr/Lars

Add classification report per class to compute_metrics
ssciwr · Apr 9, 2024 · c362702 · c362702
2 parents dc30dbc + 9ae052f
commit c362702
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/moralization/transformers_model_manager.py b/moralization/transformers_model_manager.py
@@ -19,6 +19,7 @@
 import frontmatter
 from huggingface_hub import HfApi
 import shutil
+from sklearn.metrics import classification_report
 
 
 IGNORED_LABEL = -100
@@ -318,6 +319,9 @@ 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
+            ),
         }
 
     def _set_id2label(self) -> None: