From 94105d4213063b325544f6c6ef14544f0531aff6 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Wed, 29 Jan 2025 13:37:35 +0100 Subject: [PATCH] fix: readd outlier_threshold default value MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py | 2 +- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py | 2 +- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index 3316c6c..c3273eb 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -182,7 +182,7 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): di_values = di_values.dropna() f = spy.stats.genextreme.fit(di_values) cutoff = spy.stats.genextreme.ppf( - self.freqai_info.get("outlier_threshold"), *f + self.freqai_info.get("outlier_threshold", 0.999), *f ) dk.data["DI_value_mean"] = pred_df_full["DI_values"].mean() diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py index 3ab27f7..8538e3b 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV3.py @@ -129,7 +129,7 @@ class XGBoostRegressorQuickAdapterV3(BaseRegressionModel): di_values = di_values.dropna() f = spy.stats.genextreme.fit(di_values) cutoff = spy.stats.genextreme.ppf( - self.freqai_info.get("outlier_threshold"), *f + self.freqai_info.get("outlier_threshold", 0.999), *f ) dk.data["DI_value_mean"] = pred_df_full["DI_values"].mean() diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 6464920..222b2a7 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -179,7 +179,7 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): di_values = di_values.dropna() f = spy.stats.genextreme.fit(di_values) cutoff = spy.stats.genextreme.ppf( - self.freqai_info.get("outlier_threshold"), *f + self.freqai_info.get("outlier_threshold", 0.999), *f ) dk.data["DI_value_mean"] = pred_df_full["DI_values"].mean() -- 2.43.0