text_plain_output_1.png
import numpy as np

unique, counts = np.unique(y_train, return_counts=True)

unique, counts = np.unique(y_test, return_counts=True)
print('Y Test distrubution:', dict(zip(unique, counts)))