Merge pull request #41 from qwertpi/patch-1

Corrected display_activations docstring
philipperemy · May 31, 2019 · 7a3c6d4 · 7a3c6d4
2 parents 8660d66 + 542d24c
commit 7a3c6d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keract/keract.py b/keract/keract.py
@@ -84,7 +84,7 @@ def get_activations(model, x, layer_name=None):
 
 def display_activations(activations, cmap=None, save=False):
     """
-    Plot heatmaps of activations for all filters for each layer
+    Plot the activations for each layer using matplotlib
     :param activations: dict mapping layers to corresponding activations (1, output_h, output_w, num_filters)
     :param cmap: string - a valid matplotlib colourmap to be used
     :param save: bool- if the plot should be saved