DOC fix dataset used for visualization in digits classification example (#19095)

yzhenman · web-flow · commit 280087899302 · 2021-01-04T11:04:55.000+01:00
diff --git a/examples/classification/plot_digits_classification.py b/examples/classification/plot_digits_classification.py
@@ -78,8 +78,9 @@
 # digit value in the title.
 
 _, axes = plt.subplots(nrows=1, ncols=4, figsize=(10, 3))
-for ax, image, prediction in zip(axes, digits.images, predicted):
+for ax, image, prediction in zip(axes, X_test, predicted):
     ax.set_axis_off()
+    image = image.reshape(8, 8)
     ax.imshow(image, cmap=plt.cm.gray_r, interpolation='nearest')
     ax.set_title(f'Prediction: {prediction}')
 
