mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-18 02:00:09 +08:00
chore: commit all changes
This commit is contained in:
20
types/faiss.pyi
Normal file
20
types/faiss.pyi
Normal file
@@ -0,0 +1,20 @@
|
||||
import numpy as np
|
||||
|
||||
class Index:
|
||||
def add(self, x: np.ndarray) -> None: ...
|
||||
def search(self, x: np.ndarray, k: int) -> tuple[np.ndarray, np.ndarray]: ...
|
||||
def remove_ids(self, x: np.ndarray) -> None: ...
|
||||
def add_with_ids(self, x: np.ndarray, xids: np.ndarray) -> None: ...
|
||||
ntotal: int
|
||||
|
||||
class IndexIDMap(Index):
|
||||
def __init__(self, base_index: Index) -> None: ...
|
||||
def add_with_ids(self, x: np.ndarray, xids: np.ndarray) -> None: ...
|
||||
def reconstruct(self, key: int) -> np.ndarray: ...
|
||||
|
||||
class IndexFlatL2(Index):
|
||||
def __init__(self, d: int) -> None: ...
|
||||
|
||||
def read_index(filename: str) -> Index: ...
|
||||
def write_index(index: Index, filename: str | None) -> None: ...
|
||||
def normalize_L2(x: np.ndarray) -> None: ...
|
||||
Reference in New Issue
Block a user