diff --git a/freqtrade/freqai/base_models/BaseTorchClassifier.py b/freqtrade/freqai/base_models/BasePyTorchClassifier.py similarity index 99% rename from freqtrade/freqai/base_models/BaseTorchClassifier.py rename to freqtrade/freqai/base_models/BasePyTorchClassifier.py index 1cfd742db..c08142876 100644 --- a/freqtrade/freqai/base_models/BaseTorchClassifier.py +++ b/freqtrade/freqai/base_models/BasePyTorchClassifier.py @@ -16,7 +16,7 @@ from freqtrade.freqai.data_kitchen import FreqaiDataKitchen logger = logging.getLogger(__name__) -class BaseTorchClassifier(BasePyTorchModel): +class BasePyTorchClassifier(BasePyTorchModel): """ A PyTorch implementation of a classifier. User must implement fit method diff --git a/freqtrade/freqai/base_models/BaseTorchRegressor.py b/freqtrade/freqai/base_models/BasePyTorchRegressor.py similarity index 97% rename from freqtrade/freqai/base_models/BaseTorchRegressor.py rename to freqtrade/freqai/base_models/BasePyTorchRegressor.py index baaf097ee..756853496 100644 --- a/freqtrade/freqai/base_models/BaseTorchRegressor.py +++ b/freqtrade/freqai/base_models/BasePyTorchRegressor.py @@ -13,7 +13,7 @@ from freqtrade.freqai.data_kitchen import FreqaiDataKitchen logger = logging.getLogger(__name__) -class BaseTorchRegressor(BasePyTorchModel): +class BasePyTorchRegressor(BasePyTorchModel): """ A PyTorch implementation of a regressor. User must implement fit method diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py index 16866859b..20c0b0c65 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py @@ -2,13 +2,13 @@ from typing import Any, Dict import torch -from freqtrade.freqai.base_models.BaseTorchClassifier import BaseTorchClassifier +from freqtrade.freqai.base_models.BasePyTorchClassifier import BasePyTorchClassifier from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer -class PyTorchMLPClassifier(BaseTorchClassifier): +class PyTorchMLPClassifier(BasePyTorchClassifier): """ This class implements the fit method of IFreqaiModel. in the fit method we initialize the model and trainer objects. diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py b/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py index 861d90a21..df149ffbf 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py @@ -2,13 +2,13 @@ from typing import Any, Dict import torch -from freqtrade.freqai.base_models.BaseTorchRegressor import BaseTorchRegressor +from freqtrade.freqai.base_models.BasePyTorchRegressor import BasePyTorchRegressor from freqtrade.freqai.data_kitchen import FreqaiDataKitchen from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer -class PyTorchMLPRegressor(BaseTorchRegressor): +class PyTorchMLPRegressor(BasePyTorchRegressor): """ This class implements the fit method of IFreqaiModel. in the fit method we initialize the model and trainer objects.