diff options
Diffstat (limited to 'src/python/gudhi/representations/vector_methods.py')
| -rw-r--r-- | src/python/gudhi/representations/vector_methods.py | 2 |
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 8114885e..5ea4ea48 100644 --- a/src/python/gudhi/representations/vector_methods.py +++ b/src/python/gudhi/representations/vector_methods.py @@ -175,7 +175,7 @@ class Landscape(BaseEstimator, TransformerMixin): Xfit = [] x_values = self.im_range - for i, diag in enumerate(X): + for diag in X: midpoints, heights = (diag[:, 0] + diag[:, 1]) / 2., (diag[:, 1] - diag[:, 0]) / 2. tent_functions = np.maximum(heights[None, :] - np.abs(x_values[:, None] - midpoints[None, :]), 0) tent_functions.partition(diag.shape[0] - self.num_landscapes, axis=1) |