2695924bd6605168cef51e9477faa1bacd3717f5,tests/unit_test/test_embedding.py,,test_embedding,#Any#,12

Before Change

    assert (np.abs(embed.embedding_mat[2] - \
                   (np.array(range(10)) * 0.1 + 0.1)) < 1e-6).all()

    embed = mz.datasets.toy.load_embedding()
    embed.build(term_index_input)
    assert embed.embedding_dim == 10
    assert embed.index_state == {4: 1, 2: 1, 3: 1, 1: 2, 0: 0}
    assert (embed.embedding_mat[0] == np.array([0] * 10)).all()
    assert (np.abs(embed.embedding_mat[2] - \

After Change

def test_embedding(term_index):
    embed = mz.embedding.load_from_file(mz.datasets.embeddings.EMBED_RANK)
    matrix = embed.build_matrix(term_index)
    assert matrix.shape == (len(term_index) + 1, 50)

Instances