summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMarc Glisse <marc.glisse@inria.fr>2020-11-07 11:10:09 +0100
committerMarc Glisse <marc.glisse@inria.fr>2020-11-07 11:10:09 +0100
commita81be6810159493933cbc0159feae191ae7d0cc3 (patch)
tree5c0712595779ffddb1a7a6f6c6bf4a8eb8e79a1f
parent9ea9ae8f72c26dbe37319168cfce64afa01d0fd9 (diff)
Swap clip with int conversion
This allows correct handling of infinite values.
-rw-r--r--src/python/gudhi/representations/vector_methods.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py
index 6a1f61ef..cdcb1fde 100644
--- a/src/python/gudhi/representations/vector_methods.py
+++ b/src/python/gudhi/representations/vector_methods.py
@@ -328,7 +328,7 @@ class BettiCurve(BaseEstimator, TransformerMixin):
step_x = x_values[1] - x_values[0]
for diagram in X:
- diagram_int = np.clip(np.ceil((diagram[:,:2] - self.sample_range[0]) / step_x).astype(int), 0, self.resolution)
+ diagram_int = np.clip(np.ceil((diagram[:,:2] - self.sample_range[0]) / step_x), 0, self.resolution).astype(int)
bc = np.zeros(self.resolution)
for interval in diagram_int:
bc[interval[0]:interval[1]] += 1