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 816441b6..e85fd14d 100644
--- a/src/python/gudhi/sktda/metrics.py
+++ b/src/python/gudhi/sktda/metrics.py
@@ -26,7 +26,7 @@ class SlicedWassersteinDistance(BaseEstimator, TransformerMixin):
Constructor for the SlicedWassersteinDistance class.
Attributes:
- num_directions (int): number of lines to sample uniformly from [-pi,pi] in order to approximate and speed up the distance computation (default 10).
+ num_directions (int): number of lines evenly sampled from [-pi/2,pi/2] in order to approximate and speed up the distance computation (default 10).
"""
self.num_directions = num_directions
thetas = np.linspace(-np.pi/2, np.pi/2, num=self.num_directions+1)[np.newaxis,:-1]