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
parent
commit
894b647bee
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      learnware/learnware/reuse.py

+ 1
- 1
learnware/learnware/reuse.py View File

@@ -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:


Loading…
Cancel
Save