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

add device to data convertor class doc

Browse Source
This commit is contained in:
Yinon Polak 2023-04-03 16:36:38 +03:00
parent 36a0a14a23
commit bc9454e0f9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/freqai/torch/PyTorchDataConvertor.py
View File

@ -11,7 +11,7 @@ class PyTorchDataConvertor(ABC):
def convert_x(self, df: pd.DataFrame, device: Optional[str] = None) -> Tuple[torch.Tensor, ...]:
"""
:param df: "*_features" dataframe.
:param device: cpu/gpu.
:param device: The device to use for training (e.g. 'cpu', 'cuda').
:returns: tuple of tensors.
"""
@ -19,7 +19,7 @@ class PyTorchDataConvertor(ABC):
def convert_y(self, df: pd.DataFrame, device: Optional[str] = None) -> Tuple[torch.Tensor, ...]:
"""
:param df: "*_labels" dataframe.
:param device: cpu/gpu.
:param device: The device to use for training (e.g. 'cpu', 'cuda').
:returns: tuple of tensors.
"""