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

Limit tree depth for GPU hist. (#6045)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-08-22 19:34:52 +08:00
committed by GitHub
parent b9ebbffc57
commit a144daf034
4 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python-package/xgboost/core.py
View File

@@ -40,7 +40,7 @@ class EarlyStopException(Exception):
"""
def __init__(self, best_iteration):
super(EarlyStopException, self).__init__()
super().__init__()
self.best_iteration = best_iteration

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

@@ -1025,7 +1025,7 @@ class XGBRFClassifier(XGBClassifier):
**kwargs)
def get_xgb_params(self):
params = super(XGBRFClassifier, self).get_xgb_params()
params = super().get_xgb_params()
params['num_parallel_tree'] = self.n_estimators
return params
@@ -1057,7 +1057,7 @@ class XGBRFRegressor(XGBRegressor):
reg_lambda=reg_lambda, **kwargs)
def get_xgb_params(self):
params = super(XGBRFRegressor, self).get_xgb_params()
params = super().get_xgb_params()
params['num_parallel_tree'] = self.n_estimators
return params