summaryrefslogtreecommitdiff
path: root/src/python/gudhi/sktda/metrics.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/python/gudhi/sktda/metrics.py')
-rw-r--r--src/python/gudhi/sktda/metrics.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/python/gudhi/sktda/metrics.py b/src/python/gudhi/sktda/metrics.py
index 8092f7af..816441b6 100644
--- a/src/python/gudhi/sktda/metrics.py
+++ b/src/python/gudhi/sktda/metrics.py
@@ -34,7 +34,7 @@ class SlicedWassersteinDistance(BaseEstimator, TransformerMixin):
def fit(self, X, y=None):
"""
- Fit the SlicedWassersteinDistance class on a list of persistence diagrams: persistence diagrams are projected onto the different lines. The diagrams themselves and their projections are then stored in a numpy array called **diagrams**.
+ Fit the SlicedWassersteinDistance class on a list of persistence diagrams: persistence diagrams are projected onto the different lines. The diagrams themselves and their projections are then stored in numpy arrays, called **diagrams_** and **approx_diag_**.
Parameters:
X (list of n x 2 numpy arrays): input persistence diagrams.