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]})