Fix input in layer name. (#67)

google · Jul 13, 2021 · d03e8fe · d03e8fe
1 parent 2c55b84
commit d03e8fe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ml/pc/exercises/image_classification_part1.ipynb b/ml/pc/exercises/image_classification_part1.ipynb
@@ -571,7 +571,7 @@
         "successive_feature_maps = visualization_model.predict(x)\n",
         "\n",
         "# These are the names of the layers, so can have them as part of our plot\n",
-        "layer_names = [layer.name for layer in model.layers]\n",
+        "layer_names = [layer.name for layer in model.layers[1:]]\n",
         "\n",
         "# Now let's display our representations\n",
         "for layer_name, feature_map in zip(layer_names, successive_feature_maps):\n",