text_plain_output_1.png
"""
n = 10
cols = 5
rows = np.ceil(n/cols)
fig = plt.gcf()
fig.set_size_inches(cols * n, rows * n)
for i in range(n):
  plt.subplot(rows, cols, i+1)
  plt.imshow(train_x[i])
  plt.title(train['diagnosis'][i], fontsize=40)
  plt.axis('off')
"""