alternative, low code solution to the first bonus question

[peter-amerkhanian]

print("Train:", np.unique(y_train, return_counts=True))
print("Validation:",np.unique(y_val, return_counts=True))
print("Test:", np.unique(y_test, return_counts=True))

This is a low code alternative to viewing the distributions of digits in each data set. No visual, but students seemed to like it.