├── .gitignore ├── LICENSE ├── README.md ├── annbench ├── __init__.py ├── algo │ ├── __init__.py │ ├── annoy.py │ ├── base.py │ ├── faiss_cpu.py │ ├── faiss_gpu.py │ ├── hnsw.py │ ├── proxy.py │ ├── pynndescent.py │ └── scann.py ├── dataset │ ├── __init__.py │ ├── base.py │ ├── deep1b.py │ ├── deep1m.py │ ├── proxy.py │ ├── sift1m.py │ └── siftsmall.py ├── util.py └── vis.py ├── conf ├── algo │ ├── annoy.yaml │ ├── hnsw.yaml │ ├── hnsw_faiss.yaml │ ├── ivfpq.yaml │ ├── ivfpq4bit.yaml │ ├── ivfpq_gpu.yaml │ ├── linear.yaml │ ├── linear_gpu.yaml │ ├── nsg.yaml │ ├── pq.yaml │ ├── pq4bit.yaml │ ├── pynndescent.yaml │ └── scann.yaml ├── common.yaml ├── config_download.yaml ├── config_plot.yaml ├── config_run.yaml ├── dataset │ ├── deep1b.yaml │ ├── deep1m.yaml │ ├── sift1m.yaml │ └── siftsmall.yaml └── specialized_param │ ├── deep1b_ivfpq.yaml │ ├── deep1b_ivfpq4bit.yaml │ ├── deep1m_ivfpq.yaml │ ├── deep1m_ivfpq4bit.yaml │ ├── deep1m_ivfpq_gpu.yaml │ ├── deep1m_scann.yaml │ ├── sift1m_ivfpq.yaml │ ├── sift1m_ivfpq4bit.yaml │ ├── sift1m_ivfpq_gpu.yaml │ └── sift1m_scann.yaml ├── dataset └── .gitkeep ├── download.py ├── interim └── .gitkeep ├── log └── .gitkeep ├── output └── .gitkeep ├── plot.py ├── requirements.txt ├── result_img └── .gitkeep └── run.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/README.md -------------------------------------------------------------------------------- /annbench/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/__init__.py -------------------------------------------------------------------------------- /annbench/algo/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /annbench/algo/annoy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/annoy.py -------------------------------------------------------------------------------- /annbench/algo/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/base.py -------------------------------------------------------------------------------- /annbench/algo/faiss_cpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/faiss_cpu.py -------------------------------------------------------------------------------- /annbench/algo/faiss_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/faiss_gpu.py -------------------------------------------------------------------------------- /annbench/algo/hnsw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/hnsw.py -------------------------------------------------------------------------------- /annbench/algo/proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/proxy.py -------------------------------------------------------------------------------- /annbench/algo/pynndescent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/pynndescent.py -------------------------------------------------------------------------------- /annbench/algo/scann.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/algo/scann.py -------------------------------------------------------------------------------- /annbench/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /annbench/dataset/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/base.py -------------------------------------------------------------------------------- /annbench/dataset/deep1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/deep1b.py -------------------------------------------------------------------------------- /annbench/dataset/deep1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/deep1m.py -------------------------------------------------------------------------------- /annbench/dataset/proxy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/proxy.py -------------------------------------------------------------------------------- /annbench/dataset/sift1m.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/sift1m.py -------------------------------------------------------------------------------- /annbench/dataset/siftsmall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/dataset/siftsmall.py -------------------------------------------------------------------------------- /annbench/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/util.py -------------------------------------------------------------------------------- /annbench/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/annbench/vis.py -------------------------------------------------------------------------------- /conf/algo/annoy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/annoy.yaml -------------------------------------------------------------------------------- /conf/algo/hnsw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/hnsw.yaml -------------------------------------------------------------------------------- /conf/algo/hnsw_faiss.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/hnsw_faiss.yaml -------------------------------------------------------------------------------- /conf/algo/ivfpq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/ivfpq.yaml -------------------------------------------------------------------------------- /conf/algo/ivfpq4bit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/ivfpq4bit.yaml -------------------------------------------------------------------------------- /conf/algo/ivfpq_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/ivfpq_gpu.yaml -------------------------------------------------------------------------------- /conf/algo/linear.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/linear.yaml -------------------------------------------------------------------------------- /conf/algo/linear_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/linear_gpu.yaml -------------------------------------------------------------------------------- /conf/algo/nsg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/nsg.yaml -------------------------------------------------------------------------------- /conf/algo/pq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/pq.yaml -------------------------------------------------------------------------------- /conf/algo/pq4bit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/pq4bit.yaml -------------------------------------------------------------------------------- /conf/algo/pynndescent.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/pynndescent.yaml -------------------------------------------------------------------------------- /conf/algo/scann.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/algo/scann.yaml -------------------------------------------------------------------------------- /conf/common.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/common.yaml -------------------------------------------------------------------------------- /conf/config_download.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/config_download.yaml -------------------------------------------------------------------------------- /conf/config_plot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/config_plot.yaml -------------------------------------------------------------------------------- /conf/config_run.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/config_run.yaml -------------------------------------------------------------------------------- /conf/dataset/deep1b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/dataset/deep1b.yaml -------------------------------------------------------------------------------- /conf/dataset/deep1m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/dataset/deep1m.yaml -------------------------------------------------------------------------------- /conf/dataset/sift1m.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/dataset/sift1m.yaml -------------------------------------------------------------------------------- /conf/dataset/siftsmall.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/dataset/siftsmall.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1b_ivfpq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1b_ivfpq.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1b_ivfpq4bit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1b_ivfpq4bit.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1m_ivfpq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1m_ivfpq.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1m_ivfpq4bit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1m_ivfpq4bit.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1m_ivfpq_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1m_ivfpq_gpu.yaml -------------------------------------------------------------------------------- /conf/specialized_param/deep1m_scann.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/deep1m_scann.yaml -------------------------------------------------------------------------------- /conf/specialized_param/sift1m_ivfpq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/sift1m_ivfpq.yaml -------------------------------------------------------------------------------- /conf/specialized_param/sift1m_ivfpq4bit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/sift1m_ivfpq4bit.yaml -------------------------------------------------------------------------------- /conf/specialized_param/sift1m_ivfpq_gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/sift1m_ivfpq_gpu.yaml -------------------------------------------------------------------------------- /conf/specialized_param/sift1m_scann.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/conf/specialized_param/sift1m_scann.yaml -------------------------------------------------------------------------------- /dataset/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /download.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/download.py -------------------------------------------------------------------------------- /interim/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /log/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /output/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/plot.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/requirements.txt -------------------------------------------------------------------------------- /result_img/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/matsui528/annbench/HEAD/run.py --------------------------------------------------------------------------------