annoy

1
from annoy import AnnoyIndex
2

3
a = AnnoyIndex(3, 'angular')
4
a.add_item(0, [1, 0, 0])
5
a.add_item(1, [0, 1, 0])
6
a.add_item(2, [0, 0, 1])
7
a.build(-1)
8

9
print(a.get_nns_by_item(0, 100))
10
print(a.get_nns_by_vector([1.0, 0.5, 0.5], 100))
11