fix: update learning rate default value for XGB models

zama-ai · Nov 6, 2023 · f30e4e1 · f30e4e1
1 parent 96585aa
commit f30e4e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/concrete/ml/sklearn/xgb.py b/src/concrete/ml/sklearn/xgb.py
@@ -29,7 +29,7 @@ def __init__(
         self,
         n_bits: int = 6,
         max_depth: Optional[int] = 3,
-        learning_rate: Optional[float] = 0.1,
+        learning_rate: Optional[float] = None,
         n_estimators: Optional[int] = 20,
         objective: Optional[str] = "binary:logistic",
         booster: Optional[str] = None,
@@ -233,7 +233,7 @@ def __init__(
         self,
         n_bits: int = 6,
         max_depth: Optional[int] = 3,
-        learning_rate: Optional[float] = 0.1,
+        learning_rate: Optional[float] = None,
         n_estimators: Optional[int] = 20,
         objective: Optional[str] = "reg:squarederror",
         booster: Optional[str] = None,