From 72383e2916e0e352a08aba9b6652ebd7d1f18727 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20LESCOAT?= <58120903+loic-lescoat@users.noreply.github.com> Date: Mon, 18 Oct 2021 14:48:49 -0700 Subject: [PATCH] Fixed typo: AVG_MICRO --> AVG_MACRO --- src/NERDA/models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/NERDA/models.py b/src/NERDA/models.py index 00dc3c4..12a721a 100644 --- a/src/NERDA/models.py +++ b/src/NERDA/models.py @@ -394,7 +394,7 @@ def evaluate_performance(self, dataset: dict, y_true = dataset.get('tags'), labels = self.tag_scheme, average = 'macro') - f1_macro = pd.DataFrame({'Level' : ['AVG_MICRO'], + f1_macro = pd.DataFrame({'Level' : ['AVG_MACRO'], 'F1-Score': [f1_macro[2]], 'Precision': [np.nan], 'Recall': [np.nan]})