Merge pull request #32 from kevinalexmathews/master

Add weighted average
chakki-works · Sep 30, 2020 · 5e624d7 · 5e624d7
2 parents 332eead + a82a873
commit 5e624d7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/seqeval/metrics/sequence_labeling.py b/seqeval/metrics/sequence_labeling.py
@@ -337,7 +337,7 @@ def classification_report(y_true, y_pred, digits=2, suffix=False):
     for e in pred_entities:
         d2[e[0]].add((e[1], e[2]))
 
-    last_line_heading = 'macro avg'
+    last_line_heading = 'weighted avg'
     width = max(name_width, len(last_line_heading), digits)
 
     headers = ["precision", "recall", "f1-score", "support"]
@@ -375,6 +375,12 @@ def classification_report(y_true, y_pred, digits=2, suffix=False):
                              f1_score(y_true, y_pred, suffix=suffix),
                              np.sum(s),
                              width=width, digits=digits)
+    report += row_fmt.format('macro avg',
+                             np.average(ps),
+                             np.average(rs),
+                             np.average(f1s),
+                             np.sum(s),
+                             width=width, digits=digits)
     report += row_fmt.format(last_line_heading,
                              np.average(ps, weights=s),
                              np.average(rs, weights=s),