update docs

thierrymoudiki · Aug 3, 2024 · f05db7b · f05db7b
1 parent 5e1fc88
commit f05db7b
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
@@ -16,6 +16,9 @@ class GBDT(BaseEstimator):
 
         learning_rate: float
             shrinkage rate; used for reducing the gradient step
+        
+        max_depth: int 
+            maximum tree depth
 
         rowsample: float
             subsample ratio of the training instances

diff --git a/unifiedbooster/gbdt_classification.py b/unifiedbooster/gbdt_classification.py
@@ -24,6 +24,9 @@ class GBDTClassifier(GBDT, ClassifierMixin):
 
         learning_rate: float
             shrinkage rate; used for reducing the gradient step
+        
+        max_depth: int 
+            maximum tree depth
 
         rowsample: float
             subsample ratio of the training instances

diff --git a/unifiedbooster/gbdt_regression.py b/unifiedbooster/gbdt_regression.py
@@ -24,6 +24,9 @@ class GBDTRegressor(GBDT, RegressorMixin):
 
         learning_rate: float
             shrinkage rate; used for reducing the gradient step
+        
+        max_depth: int 
+            maximum tree depth
 
         rowsample: float
             subsample ratio of the training instances