e4deccd98b73c918f582297050ecaf899e5387eb,ristretto/tests/test_cur.py,,test_cur,#,13

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, U, R = cur(A, k=k+2, index_set=False)
    relative_error = (np.linalg.norm(A - C.dot(U).dot(R)) / np.linalg.norm(A))
    assert relative_error < 1e-4

    // index_set = True
    C, U, R = cur(A, k=k+2, index_set=True)
    relative_error = (np.linalg.norm(A - A[:,C].dot(U).dot(A[R,:])) / np.linalg.norm(A))
    assert relative_error < 1e-4


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

After Change

    // index_set = False
    C, U, R = cur(A, rank=k+2, index_set=False)
    A_cur = C.dot(U).dot(R)
    assert relative_error(A, A_cur) < atol_float32

    // index_set = True
    C, U, R = cur(A, rank=k+2, index_set=True)
    A_cur = A[:, C].dot(U).dot(A[R])

Instances