From 325a3cc17f9127b72706bda081895baaca8a7466 Mon Sep 17 00:00:00 2001
From: Googol2002 <xingyuncao2002@outlook.com>
Date: Sun, 24 Sep 2023 11:42:52 +0800
Subject: [PATCH] [Fix] The axis of softmax should be -1 instead of 0.

---
 learnware/learnware/reuse.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: