From 1c11a5f0485d2bdca5295712f4607abc0fe6d6a9 Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Sun, 19 Mar 2023 18:10:57 +0200 Subject: [PATCH] improve mlp documentation --- freqtrade/freqai/prediction_models/PyTorchMLPModel.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py index c2b6c1b93..0e6b3c7bb 100644 --- a/freqtrade/freqai/prediction_models/PyTorchMLPModel.py +++ b/freqtrade/freqai/prediction_models/PyTorchMLPModel.py @@ -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