Add n_splits_

Y-oHr-N · Mar 1, 2020 · 1b01279 · 1b01279
1 parent c8a037f
commit 1b01279
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/optgbm/sklearn.py b/optgbm/sklearn.py
@@ -563,6 +563,7 @@ def fit(
 
         self.best_params_ = {**params, **self.study_.best_params}
         self._best_iteration = self.study_.user_attrs["best_iteration"]
+        self.n_splits_ = cv.get_n_splits(X, y, groups=groups)
 
         logger = logging.getLogger(__name__)
 
@@ -696,6 +697,9 @@ class OGBMClassifier(_BaseOGBMModel, ClassifierMixin):
     n_features_
         Number of features of fitted model.
 
+    n_splits_:
+        Number of cross-validation splits.
+
     study_
         Actual study.
 
@@ -863,6 +867,9 @@ class OGBMRegressor(_BaseOGBMModel, RegressorMixin):
     n_features_
         Number of features of fitted model.
 
+    n_splits_:
+        Number of cross-validation splits.
+
     study_
         Actual study.