# Get the count of each label (0 and 1) in labels print('0:', labels[labels==1].shape[0], '1:',labels[labels==0].shape[0])