Skip to content

Commit

Permalink
Fix a pydocstyle error
Browse files Browse the repository at this point in the history
  • Loading branch information
Y-oHr-N committed Mar 2, 2020
1 parent 93e6756 commit 7ebec14
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions optgbm/sklearn.py
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@ def predict(
class _BaseOGBMModel(lgb.LGBMModel):
@property
def best_index_(self) -> int:
"""Index which corresponds to the best candidate parameter setting."""
"""Get the best trial's number."""
return self.study_.best_trial.number

def __init__(
Expand Down Expand Up @@ -720,14 +720,14 @@ class OGBMClassifier(_BaseOGBMModel, ClassifierMixin):

@property
def classes_(self) -> np.ndarray:
"""Class labels."""
"""Get the class labels."""
self._check_is_fitted()

return self._classes

@property
def n_classes_(self) -> int:
"""Number of classes."""
"""Get the number of classes."""
return self._n_classes

def predict(self, X: TWO_DIM_ARRAYLIKE_TYPE) -> ONE_DIM_ARRAYLIKE_TYPE:
Expand Down

0 comments on commit 7ebec14

Please sign in to comment.