util.py 文件源码

def _kappa_helper(y_true, y_pred, weights=None):
    # weights can be None, 'linear', or 'quadratic'
    def flatten(y):
        if len(y.shape) > 1 and y.shape[1] > 1:
            y = np.argmax(y, axis=1)
            y = y.reshape(-1)
        return y

    y_true = flatten(y_true)
    y_pred = flatten(y_pred)
    return cohen_kappa_score(y_true, y_pred, weights=weights)