ezyrb.reduction.ae.AE._build_model

AE._build_model(values)[source]

Build the torch model.

Considering the number of neurons per layer (self.layers), a feed-forward NN is defined:

  • activation function from layer i>=0 to layer i+1: self.function[i]; activation function at the output layer: Identity (by default).

Parameters:

values (numpy.ndarray) – the set values one wants to reduce.