diff --git a/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py b/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py index d6be8c1df..2f6705311 100644 --- a/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py +++ b/freqtrade/freqai/prediction_models/MLPPyTorchClassifier.py @@ -1,12 +1,12 @@ from typing import Any, Dict +import torch + from freqtrade.freqai.base_models.PyTorchModelTrainer import PyTorchModelTrainer from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.prediction_models.PyTorchClassifierClassifier import PyTorchClassifier from freqtrade.freqai.prediction_models.PyTorchMLPModel import PyTorchMLPModel -import torch - class MLPPyTorchClassifier(PyTorchClassifier): """