allow users to pass 0 test data

2022-12-05 20:55:05 +01:00
parent 5826fae8ee
commit 72b1d1c9ae
3 changed files with 8 additions and 2 deletions
--- a/freqtrade/freqai/base_models/BaseTensorFlowModel.py
+++ b/freqtrade/freqai/base_models/BaseTensorFlowModel.py
@@ -23,6 +23,7 @@ class BaseTensorFlowModel(IFreqaiModel):
        if self.ft_params.get("DI_threshold", 0):
            self.ft_params["DI_threshold"] = 0
            logger.warning("DI threshold is not configured for Keras models yet. Deactivating.")
+        self.dd.model_type = 'keras'

    def train(
        self, unfiltered_df: DataFrame, pair: str, dk: FreqaiDataKitchen, **kwargs