sorting an verify parameter

time-series-machine-learning · Nov 17, 2023 · 6991cc6 · 6991cc6
1 parent 2b82b39
commit 6991cc6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tsml_eval/evaluation/storage/classifier_results.py b/tsml_eval/evaluation/storage/classifier_results.py
@@ -265,12 +265,12 @@ def calculate_statistics(self, overwrite=False):
             self.mean_auroc = roc_auc_score(
                 self.class_labels,
                 self.predictions if self.n_classes == 2 else self.probabilities,
-                average="micro",
+                average="weighted",
                 multi_class="ovr",
             )
         if self.f1_score is None or overwrite:
             self.f1_score = f1_score(
-                self.class_labels, self.predictions, average="micro"
+                self.class_labels, self.predictions, average="macro"
             )
 
     def infer_size(self, overwrite=False):