util.py 文件源码

def check_multilabel_array(array, n_labels=None, force_binary=True):
    array = check_array(array, ensure_2d=True, allow_nd=False, dtype=int)
    if n_labels is not None and array.shape[1] != n_labels:
        raise ValueError('multilabel array must have exactly %d labels' % n_labels)
    if force_binary:
        count_ones = np.count_nonzero(array == 1)
        count_zeros = np.count_nonzero(array == 0)
        if np.size(array) != count_ones + count_zeros:
            raise ValueError('multilabel array must be binary')
    return array