diff --git a/src/concrete/ml/sklearn/xgb.py b/src/concrete/ml/sklearn/xgb.py index ed294a017..c3d23b46d 100644 --- 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,