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

improve mlp documentation

Browse Source
This commit is contained in:
Yinon Polak 2023-03-19 18:10:57 +02:00
parent 903a1dc3e5
commit 1c11a5f048
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
freqtrade/freqai/prediction_models/PyTorchMLPModel.py
View File

@ -10,6 +10,9 @@ class PyTorchMLPModel(nn.Module):
""" """
A multi-layer perceptron (MLP) model implemented using PyTorch. A multi-layer perceptron (MLP) model implemented using PyTorch.
This class mainly serves as a simple example for the integration of PyTorch model's
to freqai. It is not optimized at all and should not be used for production purposes.
:param input_dim: The number of input features. This parameter specifies the number :param input_dim: The number of input features. This parameter specifies the number
of features in the input data that the MLP will use to make predictions. of features in the input data that the MLP will use to make predictions.
:param output_dim: The number of output classes. This parameter specifies the number :param output_dim: The number of output classes. This parameter specifies the number