def get_metric_range(self, pre_dist): num_classes = len(pre_dist) if num_classes < 2: num_classes = 2 return math.log2(num_classes)