Python VectorStoreの例

コード例 #1

ファイル: test_vectors.py プロジェクト: renecyranek/sense2vec

def test_borrow():
    vecs = VectorStore(128)
    good = numpy.ndarray(shape=(vecs.nr_dim,), dtype='float32')
    vecs.borrow(good)
    bad = numpy.ndarray(shape=(vecs.nr_dim+1,), dtype='float32')
    with pytest.raises(AssertionError) as excinfo:
        vecs.borrow(bad)

コード例 #2

ファイル: test_vectors.py プロジェクト: spacy-io/sense2vec

def test_add():
    vecs = VectorStore(128)
    good = numpy.ndarray(shape=(vecs.nr_dim,), dtype="float32")
    vecs.add(good)
    bad = numpy.ndarray(shape=(vecs.nr_dim + 1,), dtype="float32")
    with pytest.raises(AssertionError) as excinfo:
        vecs.add(bad)

コード例 #3

ファイル: test_vectors.py プロジェクト: yocheah/sense2vec

def test_most_similar():
    vecs = VectorStore(4)
    vecs.add(numpy.asarray([4,2,2,2], dtype='float32'))
    vecs.add(numpy.asarray([4,4,2,2], dtype='float32'))
    vecs.add(numpy.asarray([4,4,4,2], dtype='float32'))
    vecs.add(numpy.asarray([4,4,4,4], dtype='float32'))

    indices, scores = vecs.most_similar(
        numpy.asarray([4,2,2,2], dtype='float32'), 4)
    print(list(scores))
    assert list(indices) == [0,1]
    indices, scores = vecs.most_similar(
        numpy.asarray([0.1,1,1,1], dtype='float32'), 4)
    assert list(indices) == [4,3]

コード例 #4

ファイル: test_vectors.py プロジェクト: yocheah/sense2vec

def test_borrow():
    vecs = VectorStore(128)
    good = numpy.ndarray(shape=(vecs.nr_dim,), dtype='float32')
    vecs.borrow(good)
    bad = numpy.ndarray(shape=(vecs.nr_dim+1,), dtype='float32')
    with pytest.raises(AssertionError) as excinfo:
        vecs.borrow(bad)

コード例 #5

ファイル: test_vectors.py プロジェクト: spacy-io/sense2vec

def test_most_similar():
    vecs = VectorStore(4)
    vecs.add(numpy.asarray([4, 2, 2, 2], dtype="float32"))
    vecs.add(numpy.asarray([4, 4, 2, 2], dtype="float32"))
    vecs.add(numpy.asarray([4, 4, 4, 2], dtype="float32"))
    vecs.add(numpy.asarray([4, 4, 4, 4], dtype="float32"))

    indices, scores = vecs.most_similar(numpy.asarray([4, 2, 2, 2], dtype="float32"), 4)
    print(list(scores))
    assert list(indices) == [0, 1]
    indices, scores = vecs.most_similar(numpy.asarray([0.1, 1, 1, 1], dtype="float32"), 4)
    assert list(indices) == [4, 3]

コード例 #6

ファイル: test_vectors.py プロジェクト: yocheah/sense2vec

def test_init():
    vec = VectorStore(128)
    assert vec.mem is not None
    with pytest.raises(AttributeError) as excinfo:
        vec.mem = None

コード例 #7

ファイル: test_vectors.py プロジェクト: renecyranek/sense2vec

def test_init():
    vec = VectorStore(128)
    assert vec.mem is not None
    with pytest.raises(AttributeError) as excinfo:
        vec.mem = None