diff options
Diffstat (limited to 'ot/utils.py')
-rw-r--r-- | ot/utils.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/ot/utils.py b/ot/utils.py index 582c3ff..5feb4c6 100644 --- a/ot/utils.py +++ b/ot/utils.py @@ -3,6 +3,11 @@ import numpy as np from scipy.spatial.distance import cdist +def unif(n): + """ return a uniform histogram (simplex) """ + return np.ones((n,))/n + + def dist(x1,x2=None,metric='sqeuclidean'): """Compute distance between samples in x1 and x2""" if x2 is None: |