Wrapped BallTree in Cython.

Note that this wrapper does not include the BruteForceNeighbors()
function from BallTree.h

Author: thouis

scikits/learn/neighbors.py

@@ -8,7 +8,7 @@
 import numpy as np
 
 from .base import BaseEstimator, ClassifierMixin, RegressorMixin
-from .ball_tree import BallTree, knn_brute
+from .ball_tree import PyBallTree as BallTree
 from .metrics import euclidean_distances
 
 

scikits/learn/setup.py

@@ -38,7 +38,10 @@ def configuration(parent_package='', top_path=None):
         warnings.warn(BlasNotFoundError.__doc__)
 
     config.add_extension('ball_tree',
-                         sources=[join('src', 'BallTree.cpp')],
+                         sources=[join('src', 'ball_tree.cpp')],
+                         depends=[join('src', 'BallTree.h'),
+                                  join('src', 'BallTreePoint.h')],
+                         libraries=["stdc++"],
                          include_dirs=[numpy.get_include()])