e4deccd98b73c918f582297050ecaf899e5387eb,ristretto/tests/test_interp_decomp.py,,test_id_col,#,14

Before Change

    m, k = 100, 10
    A = np.array(np.random.randn(m, k), np.float64)
    A = A.dot(A.T)
    A = A[:,0:50]

    // index_set = False
    C, V = interp_decomp(A, k=k+2, mode="column", index_set=False)
    relative_error = (np.linalg.norm(A - C.dot(V)) / np.linalg.norm(A))
    assert relative_error < 1e-4

    // index_set = True
    C, V = interp_decomp(A, k=k+2, mode="column", index_set=True)
    relative_error = (np.linalg.norm(A - A[:,C].dot(V)) / np.linalg.norm(A))
    assert relative_error < 1e-4

def test_id_row():
    m, k = 100, 10

After Change

    // ------------------------------------------------------------------------
    // test index_set == False
    C, V = interp_decomp(A, rank=k+2, mode="column", index_set=False)
    A_id = C.dot(V)
    assert relative_error(A, A_id) < atol_float32

    // ------------------------------------------------------------------------
    // test index_set == True

Instances