search
接口定义
def search(q: numpy.ndarray, final_num_neighbors: int, pre_reorder_num_neighbors: int, leaves_to_search: int) -> tuple(index, distance)
接口用途
单query搜索接口，单线程执行（与开源算法接口保持一致）。
参数说明
参数名称 |
数据类型 |
描述 |
取值范围 |
|---|---|---|---|
q |
numpy.ndarray |
Query。 |
限制非空。 |
final_num_neighbors |
int |
最终返回结果数量，TopK。 |
大于等于1。 |
pre_reorder_num_neighbors |
int |
重排前保存结果数量。 |
大于等于final_num_neighbors，其中“final_num_neighbors”为最终返回结果数量。 |
leaves_to_search |
int |
搜索子空间数量。 |
[1, num_leaves]，其中“num_leaves”为IVF倒排索引分区总子空间个数。 |
返回值
数据类型 |
说明 |
|---|---|
tuple(index, distance) |
index为包含final_num_neighbors个底库ID的numpy.ndarray；distance为包含final_num_neighbors个距离的numpy.ndarray。 |
父主题： Python接口