Add best_index_

Y-oHr-N · Mar 1, 2020 · c8a037f · c8a037f
1 parent ed7bf56
commit c8a037f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/optgbm/sklearn.py b/optgbm/sklearn.py
@@ -263,6 +263,11 @@ def predict(
 
 
 class _BaseOGBMModel(lgb.LGBMModel):
+    @property
+    def best_index_(self) -> int:
+        """Index which corresponds to the best candidate parameter setting."""
+        return self.study_.best_trial.number
+
     def __init__(
         self,
         boosting_type: str = "gbdt",