From 819cc9c0e41561bedb5adb3a4fdde6aabe193137 Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 10 Jul 2022 11:06:18 +0200 Subject: [PATCH] Fully align LightGBM with Catboost --- freqtrade/freqai/prediction_models/LightGBMPredictionModel.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py b/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py index 93ded6e67..d59a0cc33 100644 --- a/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py +++ b/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py @@ -97,13 +97,11 @@ class LightGBMPredictionModel(IFreqaiModel): Filter the prediction features data and predict with it. :param: unfiltered_dataframe: Full dataframe for the current backtest period. :return: - :predictions: np.array of predictions + :pred_df: dataframe containing the predictions :do_predict: np.array of 1s and 0s to indicate places where freqai needed to remove data (NaNs) or felt uncertain about data (PCA and DI index) """ - # logger.info("--------------------Starting prediction--------------------") - dk.find_features(unfiltered_dataframe) filtered_dataframe, _ = dk.filter_features( unfiltered_dataframe, dk.training_features_list, training_filter=False