fsml.learn.models package

Submodules

fsml.learn.models.mlp module

class fsml.learn.models.mlp.FSML_MLP_Predictor(input_dimension: int, input_side_layers: int, input_hidden_size: int, output_dimension: int, output_side_layers: int, output_hidden_size: int)

Bases: Module

A simple implementation of MLP for regression

count_parameters()

forward(data: Tensor) → Tensor

The forward function for the neural network

Module contents