diff --git a/learnware/learnware/reuse.py b/learnware/learnware/reuse.py index 6252032..c81cba1 100644 --- a/learnware/learnware/reuse.py +++ b/learnware/learnware/reuse.py @@ -308,7 +308,7 @@ class AveragingReuser(BaseReuser): else: mean_pred_y += pred_y elif self.mode == "vote": - softmax_pred = softmax(pred_y, axis=0) + softmax_pred = softmax(pred_y, axis=-1) if mean_pred_y is None: mean_pred_y = softmax_pred else: