RMSprop constructor

RMSprop(
  1. List<Tensor> parameters, {
  2. required double learningRate,
  3. double beta = 0.99,
  4. double epsilon = 1e-8,
})

Implementation

RMSprop(
    List<Tensor<dynamic>> parameters, {
      required double learningRate,
      this.beta = 0.99,
      this.epsilon = 1e-8,
    }) : super(parameters, learningRate: learningRate) {
  _s = {};
  for (int p = 0; p < parameters.length; p = p + 1) {
    Tensor<dynamic> param = parameters[p];
    int size = param.data.length;
    List<double> sList = [];
    for (int i = 0; i < size; i = i + 1) {
      sList.add(0.0);
    }
    _s[param.id] = sList;
  }
}