Project - Cat vs Non-cat Classifier - Visualize the Data

Each line of train_set_x_orig and test_set_x_orig is an array representing an image. we can visualize an example by running the following code. Please feel free to change the index and re-run to see other images

# Example of a picture

index = 50
plt.imshow(train_set_x_orig[index]) # You should see a cat image
y = train_set_y[:, index]
y_class = classes[np.squeeze(train_set_y[:, index])].decode("utf-8")
print(y)
print(y_class)