rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-10 22:02:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Python] Support replace during create_index (#233)

Browse Source 
Closes #214
This commit is contained in:
Lei Xu
2023-06-27 16:02:07 -07:00
committed by GitHub
parent c68c236f17
commit 4bc676e26a
3 changed files with 40 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/lancedb/table.py
View File

@@ -188,13 +188,15 @@ class LanceTable:
num_partitions=256,
num_sub_vectors=96,
vector_column_name=VECTOR_COLUMN_NAME,
replace: bool = True,
):
"""Create an index on the table.
Parameters
----------
metric: str, default "L2"
The distance metric to use when creating the index. Valid values are "L2" or "cosine".
The distance metric to use when creating the index.
Valid values are "L2", "cosine", or "dot".
L2 is euclidean distance.
num_partitions: int
The number of IVF partitions to use when creating the index.
@@ -202,6 +204,11 @@ class LanceTable:
num_sub_vectors: int
The number of PQ sub-vectors to use when creating the index.
Default is 96.
vector_column_name: str, default "vector"
The vector column name to create the index.
replace: bool, default True
If True, replace the existing index if it exists.
If False, raise an error if duplicate index exists.
"""
self._dataset.create_index(
column=vector_column_name,
@@ -209,6 +216,7 @@ class LanceTable:
metric=metric,
num_partitions=num_partitions,
num_sub_vectors=num_sub_vectors,
replace=replace,
)
self._reset_dataset()