summaryrefslogtreecommitdiff
path: root/ot/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'ot/utils.py')
-rw-r--r--ot/utils.py5
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: