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.
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
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