TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

docs: a real example on recompute

Browse Source
This commit is contained in:
Andy Lee
2025-08-14 11:56:10 -07:00
parent 16f4572fe7
commit 79ca32e87b
2 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
benchmarks/benchmark_no_recompute.py
View File

@@ -61,12 +61,19 @@ def main():
t_recompute = bench_once(index_path_recompute, recompute=True)
t_norecompute = bench_once(index_path_norecompute, recompute=False)
size_recompute = sum(
f.stat().st_size for f in Path(index_path_recompute).parent.iterdir() if f.is_file()
)
size_norecompute = sum(
f.stat().st_size for f in Path(index_path_norecompute).parent.iterdir() if f.is_file()
)
# Compute sizes only for files belonging to each index prefix
def _size_for(prefix: str) -> int:
p = Path(prefix)
base = p.parent
stem = p.stem # e.g., 'recompute.leann'
total = 0
for f in base.iterdir():
if f.is_file() and f.name.startswith(stem):
total += f.stat().st_size
return total
size_recompute = _size_for(index_path_recompute)
size_norecompute = _size_for(index_path_norecompute)
print("Benchmark results (HNSW):")
print(