From ef4bd240c7ff397cf0cc74274012572d414f26e0 Mon Sep 17 00:00:00 2001 From: alanakbik Date: Mon, 21 Sep 2020 20:27:04 +0200 Subject: [PATCH] GH-1868: correct variable names --- flair/models/sequence_tagger_model.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/flair/models/sequence_tagger_model.py b/flair/models/sequence_tagger_model.py index 1665ec3914..f76b73167b 100644 --- a/flair/models/sequence_tagger_model.py +++ b/flair/models/sequence_tagger_model.py @@ -573,14 +573,14 @@ def evaluate( zero_division=1) # get scores - macro_f_score = round(metrics.fbeta_score(y_true, y_pred, beta=self.beta, average='micro'), 4) - micro_f_score = round(metrics.fbeta_score(y_true, y_pred, beta=self.beta, average='macro'), 4) + micro_f_score = round(metrics.fbeta_score(y_true, y_pred, beta=self.beta, average='micro'), 4) + macro_f_score = round(metrics.fbeta_score(y_true, y_pred, beta=self.beta, average='macro'), 4) accuracy_score = round(metrics.accuracy_score(y_true, y_pred), 4) detailed_result = ( "\nResults:" - f"\n- F-score (micro) {macro_f_score}" - f"\n- F-score (macro) {micro_f_score}" + f"\n- F-score (micro) {micro_f_score}" + f"\n- F-score (macro) {macro_f_score}" f"\n- Accuracy {accuracy_score}" '\n\nBy class:\n' + classification_report ) @@ -590,7 +590,7 @@ def evaluate( log_line = f"\t{accuracy_score}" result = Result( - main_score=macro_f_score, + main_score=micro_f_score, log_line=log_line, log_header=log_header, detailed_results=detailed_result,