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

Remove remaining reg:linear. (#4544)

Browse Source
This commit is contained in:
Jiaming Yuan
2019-06-11 16:04:09 +08:00
committed by GitHub
parent 4e9965cb9d
commit 4591039eba
3 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -136,11 +136,12 @@ class XGBModel(XGBModelBase):
"""
def __init__(self, max_depth=3, learning_rate=0.1, n_estimators=100,
verbosity=1, silent=None, objective="reg:linear", booster='gbtree',
n_jobs=1, nthread=None, gamma=0, min_child_weight=1,
max_delta_step=0, subsample=1, colsample_bytree=1, colsample_bylevel=1,
colsample_bynode=1, reg_alpha=0, reg_lambda=1, scale_pos_weight=1,
base_score=0.5, random_state=0, seed=None, missing=None,
verbosity=1, silent=None, objective="reg:squarederror",
booster='gbtree', n_jobs=1, nthread=None, gamma=0,
min_child_weight=1, max_delta_step=0, subsample=1,
colsample_bytree=1, colsample_bylevel=1, colsample_bynode=1,
reg_alpha=0, reg_lambda=1, scale_pos_weight=1, base_score=0.5,
random_state=0, seed=None, missing=None,
importance_type="gain", **kwargs):
if not SKLEARN_INSTALLED:
raise XGBoostError('sklearn needs to be installed in order to use this module')
@@ -377,7 +378,7 @@ class XGBModel(XGBModelBase):
if callable(self.objective):
obj = _objective_decorator(self.objective)
params["objective"] = "reg:linear"
params["objective"] = "reg:squarederror"
else:
obj = None
@@ -928,7 +929,7 @@ class XGBRFRegressor(XGBRegressor):
def __init__(self, max_depth=3, learning_rate=1, n_estimators=100,
verbosity=1, silent=None,
objective="reg:linear", n_jobs=1, nthread=None, gamma=0,
objective="reg:squarederror", n_jobs=1, nthread=None, gamma=0,
min_child_weight=1, max_delta_step=0, subsample=0.8, colsample_bytree=1,
colsample_bylevel=1, colsample_bynode=0.8, reg_alpha=0, reg_lambda=1,
scale_pos_weight=1, base_score=0.5, random_state=0, seed=None,