土法炼钢兴趣小组的算法知识备份

HNSW 最小复现 demo

源码下载

本文相关源码已整理,共 2 个文件。

打开下载目录 →

目录

HNSW 最小复现 demo

纯 numpy 实现的 HNSW 索引,用于配合文章 《向量索引深度:HNSW、DiskANN、SPANN 原理对比》 演示算法行为。

依赖

python -m venv .venv
source .venv/bin/activate
pip install numpy

运行

python bench.py

预期输出:

n=1000, d=64, nq=50, k=10
Build: 1.xx s, Query: 0.xx s (xx QPS)
Recall@10 = 0.95+

文件

调参实验

修改 bench.py 里的 HNSW(d=d, M=8, efC=100, efS=50)

此实现不做 SIMD、不加锁、不支持删除,仅用于理解算法。生产场景请使用 hnswlibFAISS


By .