diff --git a/tsml_eval/evaluation/storage/classifier_results.py b/tsml_eval/evaluation/storage/classifier_results.py index 48d36ff8..9cacd2c0 100644 --- 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):