summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/python/gudhi/representations/vector_methods.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py
index 98cd6153..77b2836f 100644
--- a/src/python/gudhi/representations/vector_methods.py
+++ b/src/python/gudhi/representations/vector_methods.py
@@ -615,16 +615,16 @@ class Atol(BaseEstimator, TransformerMixin):
>>> c = np.array([[3, 2, -1], [1, 2, -1]])
>>> atol_vectoriser = Atol(quantiser=KMeans(n_clusters=2, random_state=202006))
>>> atol_vectoriser.fit(X=[a, b, c]).centers
- array([[ 2.6 , 2.8 , -0.4 ],
- [ 2. , 0.66666667, 3.33333333]])
+ array([[ 2. , 0.66666667, 3.33333333],
+ [ 2.6 , 2.8 , -0.4 ]])
>>> atol_vectoriser(a)
- array([0.58394704, 1.0769395 ])
+ array([1.0769395 , 0.58394704])
>>> atol_vectoriser(c)
- array([1.02816136, 0.23559623])
+ array([0.23559623, 1.02816136])
>>> atol_vectoriser.transform(X=[a, b, c])
- array([[0.58394704, 1.0769395 ],
- [1.04696684, 0.56203292],
- [1.02816136, 0.23559623]])
+ array([[1.0769395 , 0.58394704],
+ [0.56203292, 1.04696684],
+ [0.23559623, 1.02816136]])
"""
def __init__(self, quantiser, weighting_method="cloud", contrast="gaussian"):
"""