WIP: Python wrapper

[grihabor]

Hi! I need some feedback on my python wrapper for your library.

Build and Install

Package uses setuptools extension interface, so it can be properly built and installed into virtualenv.
You need g++, make, cmake and swig to be installed on your system.

> python setup.py install

Try it out

from ivfhnsw import Index, IndexGrouping
import numpy as np

index = Index(4, 4, 4, 4)
index.build_quantizer('', '', '')
x = np.array([[5, 5, 5, 5]], dtype=np.float32)
labels = index.assign(x, 1)
distances, labels_ = index.search([[1, 2, 3, 4]], 1)
print(labels, distances, labels_)

This example is from python-tests/ directory

Plans

I need to know if anyone really needs a fully working python interface.
Any help is appreciated!

[dbaranchuk]

Hi,

Thanks for your contribution! I'll test your wrapper on real data at an early date. If everything is fine, I'll accept your request.

[grihabor]

I think, it's a good idea to split the pull request into 2 parts:

• code structure refactoring (now you can use cmake out-of-source)
• python wrapper

In addition to the python interface, please also check the library build