diff options
author | Marc Glisse <marc.glisse@inria.fr> | 2020-05-05 20:58:54 +0200 |
---|---|---|
committer | Marc Glisse <marc.glisse@inria.fr> | 2020-05-05 20:58:54 +0200 |
commit | 8ba3ca48e03e379fca0a0b68a508d8357a367f52 (patch) | |
tree | 60a495fe2f74ed17ce394f8f5434da2f3db43e89 | |
parent | 4c9c01ab794901210859e299c3528be8f26f5f27 (diff) |
Fix type of neighbors for "radius"
-rw-r--r-- | src/python/gudhi/clustering/tomato.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/clustering/tomato.py b/src/python/gudhi/clustering/tomato.py index 88a1a34d..1a2887bc 100644 --- a/src/python/gudhi/clustering/tomato.py +++ b/src/python/gudhi/clustering/tomato.py @@ -109,7 +109,7 @@ class Tomato: if input_type == "distance_matrix" and self.graph_type_ == "radius": X = numpy.array(X) r = self.params_["r"] - self.neighbors_ = [numpy.nonzero(l <= r) for l in X] + self.neighbors_ = [numpy.flatnonzero(l <= r) for l in X] if input_type == "distance_matrix" and self.graph_type_ == "knn": k = self.params_["k"] |