diff options
| author | Marc Glisse <marc.glisse@inria.fr> | 2021-04-15 23:46:42 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-15 23:46:42 +0200 |
| commit | 3aa34990101489b98a8bc1bd3abc55f0d77ebf7c (patch) | |
| tree | cd10514d2a41eee525b200b63a4e277f999c58a4 | |
| parent | 7e05e915adc1be285e04eb00d3ab7ba1b797f38d (diff) | |
| parent | 2ce8b315fe5e069d6444b79f6b01e76327fa5d1d (diff) | |
Merge pull request #475 from mglisse/sklearn_ngb
Name the argument n_neighbors of NearestNeighbors
| -rw-r--r-- | src/python/gudhi/point_cloud/knn.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/point_cloud/knn.py b/src/python/gudhi/point_cloud/knn.py index 994be3b6..829bf1bf 100644 --- a/src/python/gudhi/point_cloud/knn.py +++ b/src/python/gudhi/point_cloud/knn.py @@ -111,7 +111,7 @@ class KNearestNeighbors: nargs = { k: v for k, v in self.params.items() if k in {"p", "n_jobs", "metric_params", "algorithm", "leaf_size"} } - self.nn = NearestNeighbors(self.k, metric=self.metric, **nargs) + self.nn = NearestNeighbors(n_neighbors=self.k, metric=self.metric, **nargs) self.nn.fit(X) if self.params["implementation"] == "hnsw": |