diff options
| author | Marc Glisse <marc.glisse@inria.fr> | 2020-05-20 22:40:39 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-05-20 22:40:39 +0200 |
| commit | b6f798f0df407440dbaaa5f0dc9f5995e52b076e (patch) | |
| tree | 177923f5ccf9ef9b6b58ff61a30bacbbe8aa2147 /src/python/gudhi/point_cloud/knn.py | |
| parent | d7155dfcc3ed2da82569f575baacd54f7763246d (diff) | |
| parent | bb9b6b2a58d3b31a0e25d473339f2bde6430a52d (diff) | |
Merge pull request #313 from mglisse/dtmdensity
DTM density estimator
Diffstat (limited to 'src/python/gudhi/point_cloud/knn.py')
| -rw-r--r-- | src/python/gudhi/point_cloud/knn.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/python/gudhi/point_cloud/knn.py b/src/python/gudhi/point_cloud/knn.py index 86008bc3..4652fe80 100644 --- a/src/python/gudhi/point_cloud/knn.py +++ b/src/python/gudhi/point_cloud/knn.py @@ -306,6 +306,10 @@ class KNearestNeighbors: if self.params["implementation"] == "ckdtree": qargs = {key: val for key, val in self.params.items() if key in {"p", "eps", "n_jobs"}} distances, neighbors = self.kdtree.query(X, k=self.k, **qargs) + if k == 1: + # SciPy decided to squeeze the last dimension for k=1 + distances = distances[:, None] + neighbors = neighbors[:, None] if self.return_index: if self.return_distance: return neighbors, distances |