ezyrb.reduction.ae.AE.fit

AE.fit(values)[source]

Build the AE given ‘values’ and perform training.

Training procedure information:
  • optimizer: Adam’s method with default parameters (see, e.g., https://pytorch.org/docs/stable/optim.html);

  • loss: self.loss (if none, the Mean Squared Loss is set by default).

  • stopping criterion: the fulfillment of the requested tolerance on the training loss compatibly with the prescribed budget of training iterations (if type(self.stop_training) is list); if type(self.stop_training) is int or type(self.stop_training) is float, only the number of maximum iterations or the accuracy level on the training loss is considered as the stopping rule, respectively.

Parameters:

values (numpy.ndarray) – the (training) values in the points.