/rs-embed

Python module for interacting with embeddings

Primary LanguageRust

rs-embed

Python module for operations on vector embeddings.

Install

First, make sure Rust is installed. Rust nightly build is required for pyo3. Run rustup override set nightly-2019-09-01 inside the directory where the repository is cloned (tested with nightly 1.39.0). Next, run python3 setup.py install --user.

Tests

Run pytest -v tests.

Data format

There are two files: an ids file and data file. The ids file is an array of u64 ids corresponding to the vectors and the data file is an array of f32 values with total length d times the length of the ids file.