From 542d24ce06d175e98e1417df83f32293a1c94fea Mon Sep 17 00:00:00 2001 From: qwertpi Date: Thu, 30 May 2019 10:47:56 +0100 Subject: [PATCH] Corrected display_activations docstring --- keract/keract.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keract/keract.py b/keract/keract.py index 9f23935..07b34c3 100644 --- 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