nits

dnouri · Mar 11, 2016 · 3a72b8f · 3a72b8f
1 parent c7cb9e6
commit 3a72b8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nolearn/lasagne/visualize.py b/nolearn/lasagne/visualize.py
@@ -251,8 +251,8 @@ def saliency_map(input, output, pred, X):
 
 
 def saliency_map_net(net, X):
-    input = net.layers_.values()[0].input_var
-    output = get_output(net._output_layer)
+    input = net.layers_[0].input_var
+    output = get_output(net.layers_[-1])
     pred = output.eval({input: X}).argmax(axis=1)
     return saliency_map(input, output, pred, X)[0].transpose(1, 2, 0).squeeze()