- Fixed the stupid return error in predict_proba

On branch master Your branch is up to date with 'origin/master'. Changes to be committed: modified: baseline/erp/blda.py modified: baseline/erp/swlda.py
okbalefthanded · Jun 26, 2022 · e05048c · e05048c
1 parent 063bf2c
commit e05048c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/baseline/erp/blda.py b/baseline/erp/blda.py
@@ -108,5 +108,5 @@ def predict_proba(self,X):
         # return softmax(self.decision_function(X))
         # return self.decision_function(X)
         decision = self.decision_function(X)
-        return proba = expit(decision)
+        return expit(decision)
 
diff --git a/baseline/erp/swlda.py b/baseline/erp/swlda.py
@@ -48,6 +48,6 @@ def predict_proba(self,X):
         # return softmax(self.decision_function(X))
         # return self.decision_function(X)
         decision = self.decision_function(X)
-        return proba = expit(decision)
+        return expit(decision)