def calc_class_entropy(y): class_counts = stats.itemfreq(y)[:, 1] return stats.entropy(class_counts, base=2)