def distribution_class(train_dataset, test_dataset):
    labels = np.concatenate((train_dataset,test_dataset),axis=0)
    unique_labels, counts = np.unique(labels,return_counts=True)
    plt.bar(unique_labels, counts)
    plt.plot()
    plt.show()
    return unique_labels
unique_label = distribution_class(y_train, y_test)

        __
        __