Package smile.deep.optimizer

Interface Optimizer

All Superinterfaces:
Serializable
All Known Implementing Classes:
Adam, RMSProp, SGD
public interface Optimizer extends Serializable
The neural network optimizer.

  • Method Summary

    Modifier and Type
    Method
    Description
    void
    update(Layer layer, int m, int t)
    Updates a layer.

  • Method Details

    • update

      void update(Layer layer, int m, int t)
      Updates a layer.
      Parameters:
      layer - a neural network layer.
      m - the size of mini-batch.
      t - the time step, i.e. the number of training iterations so far.