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

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.