getWeights method

@override Map<String, dynamic> getWeights()

Implementation

@override
Map<String, dynamic> getWeights() {
  Map<String, dynamic> networkWeights = {};
  int layerIndex = 0;
  for (int i = 0; i < layers.length; i = i + 1) {
    Layer<dynamic, dynamic> layer = layers[i];
    if (layer.parameters.isNotEmpty) {
      networkWeights['layer_$layerIndex'] = layer.getWeights();
      layerIndex = layerIndex + 1;
    }
  }
  return networkWeights;
}