[SciPy-Dev] Distance Metrics

Jacob VanderPlas vanderplas at astro.washington.edu
Wed Jan 11 00:59:15 EST 2012


Hello,
I've been working on a little project lately centered around distance 
metrics ( https://github.com/jakevdp/pyDistances ).  The idea was to 
create a set of cython distance metrics that can be called as normal 
from python with numpy arrays, but which also expose low-level C 
function pointers so that the same metrics can be called directly on 
memory buffers from within cythonized tree-based KNN searches (KD Tree, 
Ball Tree, etc.), without any python overhead.

I initially had in mind developing this for scikit-learn in order to 
extend the capability of Ball Tree, but it occurred to me that this 
might be nice to have in scipy as well.  The speed of computing a 
distance matrix is comparable to that of pdist/cdist in 
scipy.spatial.distance (a few metrics are slightly faster, a few are 
slightly slower).  The primary advantage to this approach is the 
exposure of underlying C functions which can be easily imported and 
called from other cython scripts. 

I think there are several other advantages over the current scipy 
implementation.  Because the new code is pure cython, it would likely be 
easier to maintain and to add metrics than the current scipy setup, 
which relies on C routines wrapped by-hand using the numpy C-API.  
Because all distance functions rely on the same set of underlying cython 
routines, there are fewer places for error (for instance, currently the 
scipy.spatial.distance boolean routines return different results 
depending on whether you call the metrics directly or use cdist/pdist)

I'm curious what people think: could a framework like this replace the 
current scipy.spatial.distances implementation?  Are there any 
disadvantages that I'm not noticing?
Thanks
    Jake



More information about the SciPy-Dev mailing list