Browse Source
Merge pull request #1 from Learnware-LAMDA/dev_shihy
[Fix] The axis of softmax should be -1 instead of 0.
tags/v0.3.2
Gene
GitHub
2 years ago
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with
1 additions and
1 deletions
-
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: |
|
|
|
|