pynndescent_approximate_nearest_neighbors_index_and_query.py

1from pynndescent import NNDescent
2import numpy as np
3from sklearn.datasets import make_blobs
4
5# Generate some synthetic data
6data, _ = make_blobs(n_samples=1000, n_features=10, random_state=42)
7
8# Create the index
9index = NNDescent(data)
10
11# Query the index for the 5 nearest neighbors of the first 3 points
12indices, distances = index.query(data[:3], k=5)
13
14# Print the indices of the neighbors for the first query point
15print(indices[0])