sklearn compat

basaks · Aug 15, 2023 · 3ec8bd6 · 3ec8bd6
1 parent 535e4cb
commit 3ec8bd6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/uncoverml/optimise/models.py b/uncoverml/optimise/models.py
@@ -623,7 +623,7 @@ def __init__(self, target_transform='identity',
         self.target_transform = target_transform
         # loss = 'quantile'  # use quantile loss for median
         # alpha = 0.5  # median
-        self.median_quantile_params ={'objective': 'quantile', "metric": "quantile", 'alpha': 0.5}
+        self.median_quantile_params ={'objective': 'quantile', "metric": "quantile", 'alpha': alpha}
         self.upper_quantile_params = {'objective': 'quantile', "metric": "quantile", 'alpha': upper_alpha}
         self.lower_quantile_params = {'objective': 'quantile', "metric": "quantile", 'alpha': lower_alpha}
 
@@ -639,6 +639,7 @@ def __init__(self, target_transform='identity',
             **kwargs,
             **self.lower_quantile_params
         )
+        self.alpha = alpha
         self.upper_alpha = upper_alpha
         self.lower_alpha = lower_alpha