Merge pull request #880 from pauloalves86/master

Improve compatibility with sklearn
This commit is contained in:
Tianqi Chen 2016-02-29 08:37:53 -08:00
commit 728b65cec0

View File

@ -44,6 +44,7 @@ def _objective_decorator(func):
return func(labels, preds)
return inner
class XGBModel(XGBModelBase):
# pylint: disable=too-many-arguments, too-many-instance-attributes, invalid-name
"""Implementation of the Scikit-Learn API for XGBoost.
@ -500,7 +501,8 @@ class XGBClassifier(XGBModel, XGBClassifierBase):
fs_dict = dict(zip(keys, fs.values()))
all_features_dict = dict.fromkeys(range(0, self._features_count), 0)
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):