add self.classes_ (#83)

* add self.classes_

* only test earliest and latest version

* black

.github/workflows/pytest.yml

@@ -14,7 +14,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9, '3.10']
+        python-version: [3.8, '3.11']
 
     steps:
     - uses: actions/checkout@v2

hulearn/__init__.py

@@ -1 +1 @@
-__version__ = "0.3.4"
+__version__ = "0.3.5"

hulearn/classification/functionclassifier.py

@@ -1,5 +1,6 @@
 from sklearn.base import BaseEstimator, ClassifierMixin
 from sklearn.utils.validation import check_is_fitted
+from sklearn.utils.multiclass import unique_labels
 
 
 class FunctionClassifier(BaseEstimator, ClassifierMixin):
@@ -46,6 +47,7 @@ def fit(self, X, y):
         """
         # Run it to confirm no error happened.
         _ = self.func(X, **self.kwargs)
+        self.classes_ = unique_labels(y)
         self.fitted_ = True
         return self
 
@@ -55,6 +57,7 @@ def partial_fit(self, X, y, classes=None, sample_weight=None):
         """
         # Run it to confirm no error happened.
         _ = self.func(X, **self.kwargs)
+        self.classes_ = classes
         self.fitted_ = True
         return self
 

setup.py

@@ -68,6 +68,7 @@ def read(fname):
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
         "License :: OSI Approved :: MIT License",
         "Topic :: Scientific/Engineering",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",