setWeights method

  1. @override
void setWeights(
  1. Map<String, dynamic> networkWeights
)
override

Implementation

@override
void setWeights(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) {
      String key = 'layer_$layerIndex';
      if (networkWeights.containsKey(key)) {
        layer.setWeights(networkWeights[key] as Map<String, dynamic>);
        layerIndex = layerIndex + 1;
      }
    }
  }
}