Skip to content

GitLab

"...git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "28e0e6f0d599178c4cd3c8d186ac7064da19784f"
Commit 01c7b97d authored by Tsukasa OMOTO's avatar Tsukasa OMOTO Committed by Guolin Ke
Browse files

python-package: fix eval_at (#639) 

eval_at has been broken after e465f92a.
This commit will fix eval_at.
parent 31302bf7
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
python-package/lightgbm/sklearn.py
View file @ 01c7b97d
...@@ -340,6 +340,8 @@ class LGBMModel(LGBMModelBase): ...@@ -340,6 +340,8 @@ class LGBMModel(LGBMModelBase):
params.pop('n_estimators', None) params.pop('n_estimators', None)
if hasattr(self, 'n_classes_') and self.n_classes_ > 2: if hasattr(self, 'n_classes_') and self.n_classes_ > 2:
params['num_class'] = self.n_classes_ params['num_class'] = self.n_classes_
if hasattr(self, 'eval_at'):
params['ndcg_eval_at'] = self.eval_at
if self.fobj: if self.fobj:
params['objective'] = 'None' # objective = nullptr for unknown objective params['objective'] = 'None' # objective = nullptr for unknown objective
......
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