search
API Definition
def search(q: numpy.ndarray, final_num_neighbors: int, pre_reorder_num_neighbors: int, leaves_to_search: int) -> tuple(index, distance)
Function
Single-query search (single-thread behavior consistent with the open source algorithm).
Parameters
Parameter |
Description |
Data Type |
Value Range |
|---|---|---|---|
q |
The query. |
numpy.ndarray |
The value cannot be null. |
final_num_neighbors |
Number of returned results (Top K). |
int |
≥ 1 |
pre_reorder_num_neighbors |
Number of results saved before reordering. |
int |
≥ final_num_neighbors, where final_num_neighbors indicates the number of results returned. |
leaves_to_search |
Number of subspaces to be searched. |
int |
[1, num_leaves], where num_leaves indicates the total number of subspaces in the IVF index partition. |
Return Value
Data Type |
Description |
|---|---|
tuple(index, distance) |
index is the numpy.ndarray type data that contains final_num_neighbors base library IDs. distance is the numpy.ndarray type data that contains final_num_neighbors distances. |