Skip to content

GitLab

"python-package/vscode:/vscode.git/clone" did not exist on "f7d190aa5c021de10fd5924d782d2cdf4eee9494"
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
Show 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