:py:mod:`ForeTiS.model.mlp` =========================== .. py:module:: ForeTiS.model.mlp Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: ForeTiS.model.mlp.Mlp .. py:class:: Mlp(optuna_trial, datasets, featureset_name, optimize_featureset, pca_transform = None, current_model_name = None, batch_size = None, n_epochs = None, target_column = None) Bases: :py:obj:`ForeTiS.model._torch_model.TorchModel` Implementation of a class for a feedforward Multilayer Perceptron (MLP). See :obj:`~ForeTiS.model._base_model.BaseModel` and :obj:`~ForeTiS.model._torch_model.TorchModel` for more information on the attributes. .. py:method:: define_model() Definition of an MLP network. Architecture: - N_LAYERS of (Linear (+ ActivationFunction) (+ BatchNorm) + Dropout) - Linear output layer - Dropout layer Number of units in the first linear layer and percentage decrease after each may be fixed or optimized. .. py:method:: define_hyperparams_to_tune() See :obj:`~ForeTiS.model._base_model.BaseModel` for more information on the format. See :obj:`~ForeTiS.model._torch_model.TorchModel` for more information on hyperparameters common for all torch models.