adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename Torch to PyTorch

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/base_models/BaseTorchClassifier.py → freqtrade/freqai/base_models/BasePyTorchClassifier.py
View File

@@ -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

2
freqtrade/freqai/base_models/BaseTorchRegressor.py → freqtrade/freqai/base_models/BasePyTorchRegressor.py
View File

@@ -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

4
freqtrade/freqai/prediction_models/PyTorchMLPClassifier.py
View File

@@ -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.

4
freqtrade/freqai/prediction_models/PyTorchMLPRegressor.py
View File

@@ -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.