Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Removed trailing whitespaces and Change Error to XGBoostError

Browse Source
This commit is contained in:
Johan Manders 2015-10-14 14:17:57 +02:00
parent edf4595bc1
commit 82c2ba4c44
1 changed files with 21 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
python-package/xgboost/sklearn.py
View File

@ -230,12 +230,13 @@ class XGBModel(XGBModelBase):
evals_result = clf.evals_result() evals_result = clf.evals_result()
The variable evals_result will contain: The variable evals_result will contain:
{'validation_0': {'logloss': ['0.604835', '0.531479']}, 'validation_1': {'logloss': ['0.41965', '0.17686']}} {'validation_0': {'logloss': ['0.604835', '0.531479']},
'validation_1': {'logloss': ['0.41965', '0.17686']}}
""" """
if self.evals_result_: if self.evals_result_:
evals_result = self.evals_result_ evals_result = self.evals_result_
else: else:
raise Error('No results.') raise XGBoostError('No results.')
return evals_result return evals_result
@ -396,12 +397,13 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
evals_result = clf.evals_result() evals_result = clf.evals_result()
The variable evals_result will contain: The variable evals_result will contain:
{'validation_0': {'logloss': ['0.604835', '0.531479']}, 'validation_1': {'logloss': ['0.41965', '0.17686']}} {'validation_0': {'logloss': ['0.604835', '0.531479']},
'validation_1': {'logloss': ['0.41965', '0.17686']}}
""" """
if self.evals_result_: if self.evals_result_:
evals_result = self.evals_result_ evals_result = self.evals_result_
else: else:
raise Error('No results.') raise XGBoostError('No results.')
return evals_result return evals_result