text_plain_output_1.png
from skimage.io import imshow
from matplotlib import pyplot as plt
imshow(x_train.next()[0].astype(np.float32))
plt.show()
imshow(np.squeeze(y_train.next()[0].astype(np.float32)))
plt.show()
imshow(x_val.next()[0].astype(np.float32))
plt.show()
imshow(np.squeeze(y_val.next()[0].astype(np.float32)))
plt.show()