Skip to content

NeuralNetworks.MLP

class MLP (input_size, output_size, hidden_layers, sigmas, fourier_input_size, nb_fourier, norm, name) [source]

inhérite des propriétés de la classe Module

Cette classe fournit :

  • Un MLP entièrement configurable (dimensions, activation).
  • Option d'encodage Fourier Features sur les entrées.

Permet de construire rapidement un réseau de neurones multicouches rapidement sans connaisances profondes de pytorch.

Paramètres Type Optionnel Default value
input_size int Oui 1
output_size int Oui 1
hidden_layers list[int] Oui [1]
sigmas list[float] Oui None
fourier_input_size int Oui 2
nb_fourier int Oui 8
norm norm Oui 'Relu'
name str Oui 'Net'
Example
MLP_exemple.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
from NeuralNetworks import MLP

net = MLP(
    input_size    = 2,
    hidden_layers = [512,512,512,512,512,512,512,512,512,512],
    output_size   = 3,
    nb_fourier    = 256,
    sigmas        = [0.1,1],
    norm          = "Relu",
    name          = "Net"
)