- Added forgotten attributes

okbalefthanded · Jun 26, 2022 · c6fb31c · c6fb31c
1 parent dbedb2c
commit c6fb31c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/baseline/erp/blda.py b/baseline/erp/blda.py
@@ -12,6 +12,8 @@ def __init__(self, verbose=False):
         self.beta = 0
         self.w = []
         self.p = []
+        self.pos_class = 1.
+        self.neg_class = 0.
         self.verbose = verbose
 
     def fit(self, X, y=None):