gnes.indexer.chunk.annoy module¶
-
class
gnes.indexer.chunk.annoy.AnnoyIndexer(num_dim, data_path, metric='angular', n_trees=10, *args, **kwargs)[source]¶ Bases:
gnes.indexer.base.BaseChunkIndexerInitialize an AnnoyIndexer
Parameters: - num_dim (
int) – when set to -1, then num_dim is auto decided on first .add() - data_path (
str) – index data file managed by the annoy indexer - metric (
str) – - n_trees (
int) – - args –
- kwargs –
-
add(keys, vectors, weights, *args, **kwargs)[source]¶ adding new chunks and their vector representations
Parameters: - keys (
List[Tuple[int,Any]]) – list of (doc_id, offset) tuple - vectors (
ndarray) – vector representations - weights (
List[float]) – weight of the chunks
- keys (
-
train(*args, **kwargs)¶ Train the model, need to be overrided
- num_dim (