Changes to pylab code, seems to have changed API?

aigamedev · Jun 24, 2015 · 81470c0 · 81470c0
1 parent 1de7d2b
commit 81470c0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/plot_digits.py b/examples/plot_digits.py
@@ -36,15 +36,15 @@
 
 
 # Show some training images and some test images too.
-import pylab
+import matplotlib.pyplot as pylab
 
-for index, (image, label) in enumerate(zip(digits.images, digits.target)[:6]):
+for index, (image, label) in enumerate(zip(digits.images[:6], digits.target[:6])):
     pylab.subplot(2, 6, index + 1)
     pylab.axis('off')
     pylab.imshow(image, cmap=pylab.cm.gray_r, interpolation='nearest')
     pylab.title('Training: %i' % label)
 
-for index, (image, prediction) in enumerate(zip(X_test, y_pred)[:6]):
+for index, (image, prediction) in enumerate(zip(X_test[:6], y_pred[:6])):
     pylab.subplot(2, 6, index + 7)
     pylab.axis('off')
     pylab.imshow(image.reshape((8,8)), cmap=pylab.cm.gray_r, interpolation='nearest')