Add n_classes_

Y-oHr-N · Mar 1, 2020 · 4e25975 · 4e25975
1 parent 1b01279
commit 4e25975
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/optgbm/sklearn.py b/optgbm/sklearn.py
@@ -722,6 +722,11 @@ def classes_(self) -> np.ndarray:
 
         return self._classes
 
+    @property
+    def n_classes_(self) -> int:
+        """Number of classes."""
+        return self._n_classes
+
     def predict(self, X: TWO_DIM_ARRAYLIKE_TYPE) -> ONE_DIM_ARRAYLIKE_TYPE:
         """Predict using the fitted model.