From 3ec8bd641c61a174b6067f85793d07864f7ba56b Mon Sep 17 00:00:00 2001 From: Sudipta Basak Date: Tue, 15 Aug 2023 12:37:33 +1000 Subject: [PATCH] sklearn compat --- uncoverml/optimise/models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/uncoverml/optimise/models.py b/uncoverml/optimise/models.py index 277d53be..7cdb277a 100644 --- 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