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

add documentation to pytorch data convertor

Browse Source
This commit is contained in:
Yinon Polak 2023-04-03 16:39:49 +03:00
parent bc9454e0f9
commit 7b494c8333
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -31,6 +31,12 @@ class DefaultPyTorchDataConvertor(PyTorchDataConvertor):
target_tensor_type: Optional[torch.dtype] = None, target_tensor_type: Optional[torch.dtype] = None,
squeeze_target_tensor: bool = False squeeze_target_tensor: bool = False
): ):
"""
:param target_tensor_type: type of target tensor, for classification use
torch.long, for regressor use torch.float or torch.double.
:param squeeze_target_tensor: controls the target shape, used for loss functions
that requires 0D or 1D.
"""
self._target_tensor_type = target_tensor_type self._target_tensor_type = target_tensor_type
self._squeeze_target_tensor = squeeze_target_tensor self._squeeze_target_tensor = squeeze_target_tensor