From bcddd5be9747a6cdf9a33ddc9e82e6fd961578a6 Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Fri, 14 Mar 2025 15:24:44 +0100 Subject: [PATCH] refactor(qav3): use iloc everywhere MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- .../freqaimodels/LightGBMRegressorQuickAdapterV35.py | 4 +++- .../user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py index 1cf89f2..641a23e 100644 --- a/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/LightGBMRegressorQuickAdapterV35.py @@ -174,7 +174,9 @@ class LightGBMRegressorQuickAdapterV35(BaseRegressionModel): warmed_up = False pred_df_full = ( - self.dd.historic_predictions[pair].tail(num_candles).reset_index(drop=True) + self.dd.historic_predictions[pair] + .iloc[-num_candles:] + .reset_index(drop=True) ) if not warmed_up: diff --git a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py index 0e65884..f616f52 100644 --- a/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py +++ b/quickadapter/user_data/freqaimodels/XGBoostRegressorQuickAdapterV35.py @@ -177,7 +177,9 @@ class XGBoostRegressorQuickAdapterV35(BaseRegressionModel): warmed_up = False pred_df_full = ( - self.dd.historic_predictions[pair].tail(num_candles).reset_index(drop=True) + self.dd.historic_predictions[pair] + .iloc[-num_candles:] + .reset_index(drop=True) ) if not warmed_up: -- 2.43.0