update docs Pt.3

thierrymoudiki · Aug 2, 2024 · 037fd12 · 037fd12
1 parent f1d5a66
commit 037fd12
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git 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
@@ -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
@@ -32,6 +32,9 @@ class GBDTRegressor(GBDT, RegressorMixin):
         seed: int
             reproducibility seed
 
+        **kwargs: dict
+            additional parameters to be passed to the class
+
     Examples:
 
     ```python