feat: k-means tree

Signed-off-by: usamoi <[email protected]>
tensorchord · Nov 14, 2024 · cdbdc39 · cdbdc39
1 parent 1e33fec
commit cdbdc39
Show file tree

Hide file tree

Showing 11 changed files with 253 additions and 225 deletions.
diff --git a/README.md b/README.md
@@ -57,7 +57,7 @@ To create the VectorChord RaBitQ(vchordrq) index, you can use the following SQL.
 CREATE INDEX ON gist_train USING vchordrq (embedding vector_l2_ops) WITH (options = $$
 residual_quantization = true
 [build.internal]
-lists = 4096
+lists = [4096]
 spherical_centroids = false
 $$);
 ```

diff --git a/scripts/index.py b/scripts/index.py
@@ -71,7 +71,7 @@ def get_ivf_ops_config(metric, workers, k=None, name=None):
         config = "residual_quantization = true"
         internal_centroids_cfg = f"""
         [build.internal]
-        lists = {k}
+        lists = [{k}]
         build_threads = {workers}
         spherical_centroids = false
         """
@@ -80,7 +80,7 @@ def get_ivf_ops_config(metric, workers, k=None, name=None):
         config = "residual_quantization = false"
         internal_centroids_cfg = f"""
         [build.internal]
-        lists = {k}
+        lists = [{k}]
         build_threads = {workers}
         spherical_centroids = true
         """
@@ -89,7 +89,7 @@ def get_ivf_ops_config(metric, workers, k=None, name=None):
         config = "residual_quantization = false"
         internal_centroids_cfg = f"""
         [build.internal]
-        lists = {k}
+        lists = [{k}]
         build_threads = {workers}
         spherical_centroids = true
         """