Add attribute to method in BCNClassifier

BCN/BCNClassifier.py

@@ -140,7 +140,8 @@ def fit(self, X, y, **kwargs):
             Target values.
 
     """
-
+    self.classes_ = np.unique(y)
+    self.n_classes_ = len(self.classes_)
     # cf. https://rpy2.github.io/doc/latest/html/numpy.html
     # Create a converter that starts with rpy2's default converter
     # to which the numpy conversion rules are added.
@@ -165,7 +166,7 @@ def fit(self, X, y, **kwargs):
                        verbose = self.verbose,
                        show_progress = self.show_progress,
                        seed = self.seed                       
-                       )
+                       )    
     return self
 
   def predict_proba(self, X): 

HISTORY.rst

@@ -2,6 +2,11 @@
 History
 =======
 
+0.7.1 (2024-02-06)
+------------------
+
+* Add `classes_` attribute to `fit` method in BCNClassifier
+
 0.6.2 (2024-01-25)
 ------------------
 

setup.py

@@ -132,6 +132,6 @@ def install_r():
     test_suite='tests',
     tests_require=test_requirements,
     url='https://github.com/Techtonique/bcn_python',
-    version='0.7.0',
+    version='0.7.1',
     zip_safe=False,
 )