├── .gitignore ├── CMakeLists.txt ├── LICENSE ├── README.md ├── data └── README.md ├── inc ├── defines.hpp ├── index │ ├── Cluster.hpp │ ├── HASearcher.hpp │ ├── IVF.hpp │ ├── Initializer.hpp │ ├── Pool.hpp │ ├── Quantizer.hpp │ ├── Rotator.hpp │ ├── Searcher.hpp │ └── fastscan │ │ ├── FastScan.hpp │ │ └── pack_codes.hpp ├── third │ └── README.md └── utils │ ├── IO.hpp │ ├── StopW.hpp │ ├── memory.hpp │ ├── space.hpp │ └── tools.hpp ├── python ├── compute_gt.py ├── cvt_data.py ├── download_dataset.py ├── ivf.py ├── requirements.txt └── utils │ ├── io.py │ └── preprocess.py ├── results └── exrabitq │ ├── README.md │ └── openai1536_exhaf4.csv └── src ├── CMakeLists.txt ├── create_index.cpp └── test_search.cpp /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/README.md -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/data/README.md -------------------------------------------------------------------------------- /inc/defines.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/defines.hpp -------------------------------------------------------------------------------- /inc/index/Cluster.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Cluster.hpp -------------------------------------------------------------------------------- /inc/index/HASearcher.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/HASearcher.hpp -------------------------------------------------------------------------------- /inc/index/IVF.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/IVF.hpp -------------------------------------------------------------------------------- /inc/index/Initializer.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Initializer.hpp -------------------------------------------------------------------------------- /inc/index/Pool.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Pool.hpp -------------------------------------------------------------------------------- /inc/index/Quantizer.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Quantizer.hpp -------------------------------------------------------------------------------- /inc/index/Rotator.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Rotator.hpp -------------------------------------------------------------------------------- /inc/index/Searcher.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/Searcher.hpp -------------------------------------------------------------------------------- /inc/index/fastscan/FastScan.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/fastscan/FastScan.hpp -------------------------------------------------------------------------------- /inc/index/fastscan/pack_codes.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/index/fastscan/pack_codes.hpp -------------------------------------------------------------------------------- /inc/third/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/third/README.md -------------------------------------------------------------------------------- /inc/utils/IO.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/utils/IO.hpp -------------------------------------------------------------------------------- /inc/utils/StopW.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/utils/StopW.hpp -------------------------------------------------------------------------------- /inc/utils/memory.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/utils/memory.hpp -------------------------------------------------------------------------------- /inc/utils/space.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/utils/space.hpp -------------------------------------------------------------------------------- /inc/utils/tools.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/inc/utils/tools.hpp -------------------------------------------------------------------------------- /python/compute_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/compute_gt.py -------------------------------------------------------------------------------- /python/cvt_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/cvt_data.py -------------------------------------------------------------------------------- /python/download_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/download_dataset.py -------------------------------------------------------------------------------- /python/ivf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/ivf.py -------------------------------------------------------------------------------- /python/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/requirements.txt -------------------------------------------------------------------------------- /python/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/utils/io.py -------------------------------------------------------------------------------- /python/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/python/utils/preprocess.py -------------------------------------------------------------------------------- /results/exrabitq/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/results/exrabitq/README.md -------------------------------------------------------------------------------- /results/exrabitq/openai1536_exhaf4.csv: -------------------------------------------------------------------------------- 1 | nprobe,QPS,recall,ratio 2 | 5,1237.09,0.65296,1.01544 3 | -------------------------------------------------------------------------------- /src/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/src/CMakeLists.txt -------------------------------------------------------------------------------- /src/create_index.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/src/create_index.cpp -------------------------------------------------------------------------------- /src/test_search.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VectorDB-NTU/Extended-RaBitQ/HEAD/src/test_search.cpp --------------------------------------------------------------------------------