diff options
author | martinroyer <16647869+martinroyer@users.noreply.github.com> | 2020-06-09 09:42:40 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-06-09 09:42:40 +0200 |
commit | 588e7127d1616e40bf7e3de7e7797b54aee137da (patch) | |
tree | 38c7c35980855cbb7d04a5d5f987d518db851060 /src/python | |
parent | 5d5a2b1a3b6a2d3d2dc061e6e9c940677e782c80 (diff) |
tweak test results from doc
Diffstat (limited to 'src/python')
-rw-r--r-- | src/python/gudhi/representations/vector_methods.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py index 16e91812..d3b85636 100644 --- a/src/python/gudhi/representations/vector_methods.py +++ b/src/python/gudhi/representations/vector_methods.py @@ -614,18 +614,17 @@ class Atol(BaseEstimator, TransformerMixin): >>> b = np.array([[4, 2, 0], [4, 4, 0], [4, 0, 2]]) >>> c = np.array([[3, 2, -1], [1, 2, -1]]) >>> atol_vectoriser = Atol(quantiser=KMeans(n_clusters=2)) - >>> atol_vectoriser.fit(X=[a, b, c]) - >>> atol_vectoriser.centers + >>> atol_vectoriser.fit(X=[a, b, c]).centers array([[ 2.6 , 2.8 , -0.4 ], - [ 2. , 0.66666667, 3.33333333]]) + [ 2. , 0.66666667, 3.33333333]]) >>> atol_vectoriser(a) array([0.58394704, 1.0769395 ]) >>> atol_vectoriser(c) array([1.02816136, 0.23559623]) >>> atol_vectoriser.transform(X=[a, b, c]) array([[0.58394704, 1.0769395 ], - [1.04696684, 0.56203292], - [1.02816136, 0.23559623]]) + [1.04696684, 0.56203292], + [1.02816136, 0.23559623]]) """ def __init__(self, quantiser, weighting_method="cloud", contrast="gaus"): """ |