diff --git a/unifiedbooster/gbdt.py b/unifiedbooster/gbdt.py index f958dc0..36f2d43 100644 --- a/unifiedbooster/gbdt.py +++ b/unifiedbooster/gbdt.py @@ -24,6 +24,9 @@ class GBDT(BaseEstimator): seed: int reproducibility seed + + **kwargs: dict + additional parameters to be passed to the class """ def __init__( diff --git a/unifiedbooster/gbdt_classification.py b/unifiedbooster/gbdt_classification.py index f4c693d..6ff158f 100644 --- a/unifiedbooster/gbdt_classification.py +++ b/unifiedbooster/gbdt_classification.py @@ -32,6 +32,9 @@ class GBDTClassifier(GBDT, ClassifierMixin): seed: int reproducibility seed + **kwargs: dict + additional parameters to be passed to the class + Examples: ```python diff --git a/unifiedbooster/gbdt_regression.py b/unifiedbooster/gbdt_regression.py index 18e19f7..9250f52 100644 --- a/unifiedbooster/gbdt_regression.py +++ b/unifiedbooster/gbdt_regression.py @@ -32,6 +32,9 @@ class GBDTRegressor(GBDT, RegressorMixin): seed: int reproducibility seed + **kwargs: dict + additional parameters to be passed to the class + Examples: ```python