grouped_fe.chunked_knn_indices

Guide & worked example · Source

grouped_fe.chunked_knn_indices(
    embeddings,
    n_neighbors,
    block_size=2048,
    include_self=False,
)

Compute exact KNN indices and squared distances in chunks.