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

Merge branch 'master' into sklearn

Browse Source
This commit is contained in:
Yuan (Terry) Tang 2016-02-29 10:50:55 -06:00
commit fdd520d774
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -44,6 +44,7 @@ def _objective_decorator(func):
return func(labels, preds) return func(labels, preds)
return inner return inner
class XGBModel(XGBModelBase): class XGBModel(XGBModelBase):
# pylint: disable=too-many-arguments, too-many-instance-attributes, invalid-name # pylint: disable=too-many-arguments, too-many-instance-attributes, invalid-name
"""Implementation of the Scikit-Learn API for XGBoost. """Implementation of the Scikit-Learn API for XGBoost.
@ -500,7 +501,8 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
fs_dict = dict(zip(keys, fs.values())) fs_dict = dict(zip(keys, fs.values()))
all_features_dict = dict.fromkeys(range(0, self._features_count), 0) all_features_dict = dict.fromkeys(range(0, self._features_count), 0)
all_features_dict.update(fs_dict) all_features_dict.update(fs_dict)
return np.array(all_features_dict.values()) all_features = np.fromiter(all_features_dict.values(), np.float32)
return all_features / all_features.sum()
class XGBRegressor(XGBModel, XGBRegressorBase): class XGBRegressor(XGBModel, XGBRegressorBase):