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

DOC: Add docstrings to user-facing classes.

Browse Source
This commit is contained in:
Skipper Seabold 2015-04-27 11:01:46 -05:00
parent abcc09286c
commit efdbec4d4c
1 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
wrapper/xgboost.py
View File

@ -859,6 +859,10 @@ class XGBModel(XGBModelBase):
class XGBClassifier(XGBModel, XGBClassifier):
__doc__ = """
Implementation of the scikit-learn API for XGBoost classification
""" + "\n".join(XGBModel.__doc__.split('\n')[2:])
def __init__(self, max_depth=3, learning_rate=0.1, n_estimators=100, silent=True, objective="binary:logistic",
nthread=-1, gamma=0, min_child_weight=1, max_delta_step=0, subsample=1, colsample_bytree=1,
base_score=0.5, seed=0):
@ -910,4 +914,8 @@ class XGBClassifier(XGBModel, XGBClassifier):
class XGBRegressor(XGBModel, XGBRegressor):
__doc__ = """
Implementation of the scikit-learn API for XGBoost regression
""" + "\n".join(XGBModel.__doc__.split('\n')[2:])
pass