e4deccd98b73c918f582297050ecaf899e5387eb,ristretto/tests/test_interp_decomp.py,,test_rid_row,#,67

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
    Z, R = rinterp_decomp(A, k=k+2, mode="row", index_set=False)
    relative_error = (np.linalg.norm(A - Z.dot(R)) / np.linalg.norm(A))
    assert relative_error < 1e-4

    // index_set = True
    Z, R = rinterp_decomp(A, k=k+2, mode="row", index_set=True)
    relative_error = (np.linalg.norm(A - Z.dot(A[R,:])) / np.linalg.norm(A))
    assert relative_error < 1e-4


// =============================================================================

After Change

    // test index_set == False
    Z, R = rinterp_decomp(A, k+2, mode="row", index_set=False)
    A_id = Z.dot(R)
    assert relative_error(A, A_id) < atol_float32

    // ------------------------------------------------------------------------
    // test index_set == True
    Z, R = rinterp_decomp(A, k+2, mode="row", index_set=True)

Instances