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/prediction_models/CNNPredictionModel.py
+++ b/freqtrade/freqai/prediction_models/CNNPredictionModel.py
@@ -75,11 +75,16 @@ class CNNPredictionModel(BaseTensorFlowModel):
            metrics=[tf.metrics.MeanAbsoluteError()],
        )

+        if self.freqai_info.get('data_split_parameters', {}).get('test_size', 0.1) == 0:
+            val_data = None
+        else:
+            val_data = w1.val
+
        model.fit(
            w1.train,
            epochs=MAX_EPOCHS,
            shuffle=False,
-            validation_data=w1.val,
+            validation_data=val_data,
            callbacks=[early_stopping],
            verbose=1,
        )