rename Torch to PyTorch
This commit is contained in:
parent
f81e3d8667
commit
479aafc331
@ -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
|
@ -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
|
@ -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.
|
||||||
|
@ -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.
|
||||||
|
Loading…
Reference in New Issue
Block a user