rename Torch to PyTorch

This commit is contained in:
Yinon Polak 2023-03-22 17:50:00 +02:00
parent f81e3d8667
commit 479aafc331
4 changed files with 6 additions and 6 deletions

View File

@ -16,7 +16,7 @@ from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class BaseTorchClassifier(BasePyTorchModel): class BasePyTorchClassifier(BasePyTorchModel):
""" """
A PyTorch implementation of a classifier. A PyTorch implementation of a classifier.
User must implement fit method User must implement fit method

View File

@ -13,7 +13,7 @@ from freqtrade.freqai.data_kitchen import FreqaiDataKitchen
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class BaseTorchRegressor(BasePyTorchModel): class BasePyTorchRegressor(BasePyTorchModel):
""" """
A PyTorch implementation of a regressor. A PyTorch implementation of a regressor.
User must implement fit method User must implement fit method

View File

@ -2,13 +2,13 @@ from typing import Any, Dict
import torch 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.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel
from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer
class PyTorchMLPClassifier(BaseTorchClassifier): class PyTorchMLPClassifier(BasePyTorchClassifier):
""" """
This class implements the fit method of IFreqaiModel. This class implements the fit method of IFreqaiModel.
in the fit method we initialize the model and trainer objects. in the fit method we initialize the model and trainer objects.

View File

@ -2,13 +2,13 @@ from typing import Any, Dict
import torch 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.data_kitchen import FreqaiDataKitchen
from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel from freqtrade.freqai.torch.PyTorchMLPModel import PyTorchMLPModel
from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer from freqtrade.freqai.torch.PyTorchModelTrainer import PyTorchModelTrainer
class PyTorchMLPRegressor(BaseTorchRegressor): class PyTorchMLPRegressor(BasePyTorchRegressor):
""" """
This class implements the fit method of IFreqaiModel. This class implements the fit method of IFreqaiModel.
in the fit method we initialize the model and trainer objects. in the fit method we initialize the model and trainer objects.