KGN is a Nearest Neighbor Search algorithm, based on NSG
The project is currently closed-source, and we will consider open sourcing it after the associated research paper has been accepted.
sudo apt install -y git cmake g++ python3 python3-setuptools python3-pip libblas-dev liblapack-dev
pip3 install wheel pybind11 faiss-cpu
git clone https://github.com/Henry-yan/kgn.git
pip3 install kgn/pykgn-1.0.0-cp310-cp310-linux_x86_64.whl
import pykgn
The test results from running ann-benchmarks on an AWS r6i.2xlarge instance are as follows:
Recall@10 | QPS |
---|---|
0.997 | 10965.896 |
0.943 | 31347.704 |
0.918 | 36244.311 |
0.867 | 42496.601 |
0.748 | 56619.339 |
Recall@10 | QPS |
---|---|
1.000 | 8133.449 |
0.999 | 13661.276 |
0.996 | 21297.467 |
0.987 | 26630.163 |
0.932 | 38914.229 |
Recall@10 | QPS |
---|---|
0.972 | 2343.888 |
0.900 | 9411.566 |
0.885 | 11348.394 |
0.787 | 21024.987 |
0.633 | 35630.735 |
Recall@10 | QPS |
---|---|
0.971 | 2430.810 |
0.926 | 7896.176 |
0.850 | 24092.056 |
0.793 | 34103.748 |
0.697 | 46448.138 |