bugfix skip test split when empty

2023-03-28 14:40:23 +03:00
parent 8903ba5d89
commit 026b6a39a9
5 changed files with 28 additions and 15 deletions
--- a/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
+++ b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
@@ -76,5 +76,5 @@ class PyTorchMLPClassifier(BasePyTorchClassifier):
            squeeze_target_tensor=True,
            **self.trainer_kwargs,
        )
-        trainer.fit(data_dictionary)
+        trainer.fit(data_dictionary, self.splits)
        return trainer
--- a/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py
+++ b/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py
@@ -72,5 +72,5 @@ class PyTorchMLPRegressor(BasePyTorchRegressor):
            target_tensor_type=torch.float,
            **self.trainer_kwargs,
        )
-        trainer.fit(data_dictionary)
+        trainer.fit(data_dictionary, self.splits)
        return trainer