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.BaseChunkIndexer
Initialize 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 (