more metrics

Riccorl · Jan 30, 2021 · b262c32 · b262c32
1 parent e62c460
commit b262c32
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/transformer_srl/models.py b/transformer_srl/models.py
@@ -305,10 +305,10 @@ def get_metrics(self, reset: bool = False):
             metric_dict_filtered = {
                 x.split("-")[0] + "_role": y
                 for x, y in metric_dict.items()
-                if "overall" in x and "f1" in x
+                if "overall" in x #and "f1" in x
             }
             frame_metric_dict = {
-                x + "_frame": y for x, y in frame_metric_dict.items() if "fscore" in x
+                x + "_frame": y for x, y in frame_metric_dict.items() #if "fscore" in x
             }
             return {**metric_dict_filtered, **frame_metric_dict}