epfl-lts2 / pygsp

Graph Signal Processing in Python
https://pygsp.rtfd.io
BSD 3-Clause "New" or "Revised" License
483 stars 93 forks source link

Radius NN with cKDTree #60

Open gcuendet opened 4 years ago

gcuendet commented 4 years ago

Regarding my comment in PR #43 (see https://github.com/epfl-lts2/pygsp/pull/43#discussion_r336029429), I was thinking about something like that: when querying with a radius, use tree.query_ball_point instead of query such that you don't allocate a distances matrix of size NxN.

I ran the test_nearest_neighbor locally and it seems it is passing.