Fixed bug in lime_image where top_labels=None

marcotcr · Jul 11, 2017 · 9ab754b · 9ab754b
1 parent 97a1e2d
commit 9ab754b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lime/lime_image.py b/lime/lime_image.py
@@ -170,6 +170,8 @@ def explain_instance(self, image, classifier_fn, labels=(1,),
         else:
             fudged_image[:] = hide_color
 
+        top = labels
+
         data, labels = self.data_labels(image, fudged_image, segments,
                                         classifier_fn, num_samples,
                                         batch_size=batch_size)