From f30e4e10beb181f33b730697c43faec02c412565 Mon Sep 17 00:00:00 2001 From: Roman Bredehoft Date: Mon, 6 Nov 2023 13:53:20 +0100 Subject: [PATCH] fix: update learning rate default value for XGB models --- src/concrete/ml/sklearn/xgb.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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,