test_loss_functions.py 文件源码

def generate_binary_data(n_rows = 1000000, n_cols = 20):
    X = np.random.randint(low=0, high=2, size=(n_rows, n_cols))
    Y = np.random.randint(low=0, high=2, size=(n_rows, 1))
    pos_ind = Y == 1
    Y[~pos_ind] = -1
    Z = X * Y
    Z = np.require(Z, requirements=['F'], dtype=np.float64)
    return Z