Skip to content

GitLab

Unverified Commit af82fc95 authored by James Lamb's avatar James Lamb Committed by GitHub
Browse files

[python-package] change types in sklearn.py initializations (#5718)

parent f975d3fa
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
python-package/lightgbm/sklearn.py
View file @ af82fc95
...@@ -525,14 +525,14 @@ class LGBMModel(_LGBMModelBase): ...@@ -525,14 +525,14 @@ class LGBMModel(_LGBMModelBase):
self._Booster: Optional[Booster] = None self._Booster: Optional[Booster] = None
self._evals_result: _EvalResultDict = {} self._evals_result: _EvalResultDict = {}
self._best_score: _LGBM_BoosterBestScoreType = {} self._best_score: _LGBM_BoosterBestScoreType = {}
self._best_iteration: Optional[int] = None self._best_iteration: int = -1
self._other_params: Dict[str, Any] = {} self._other_params: Dict[str, Any] = {}
self._objective = objective self._objective = objective
self.class_weight = class_weight self.class_weight = class_weight
self._class_weight: Optional[Union[Dict, str]] = None self._class_weight: Optional[Union[Dict, str]] = None
self._class_map: Optional[Dict[int, int]] = None self._class_map: Optional[Dict[int, int]] = None
self._n_features = None self._n_features: int = -1
self._n_features_in = None self._n_features_in: int = -1
self._classes = None self._classes = None
self._n_classes: Optional[int] = None self._n_classes: Optional[int] = None
self.set_params(**kwargs) self.set_params(**kwargs)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment