improve mlp documentation

This commit is contained in:
Yinon Polak 2023-03-19 18:10:57 +02:00
parent 903a1dc3e5
commit 1c11a5f048

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