summaryrefslogtreecommitdiff
path: root/ot/lp/EMD.h
diff options
context:
space:
mode:
Diffstat (limited to 'ot/lp/EMD.h')
-rw-r--r--ot/lp/EMD.h5
1 files changed, 3 insertions, 2 deletions
diff --git a/ot/lp/EMD.h b/ot/lp/EMD.h
index aa92441..f42e222 100644
--- a/ot/lp/EMD.h
+++ b/ot/lp/EMD.h
@@ -26,9 +26,10 @@ typedef unsigned int node_id_type;
enum ProblemType {
INFEASIBLE,
OPTIMAL,
- UNBOUNDED
+ UNBOUNDED,
+ MAX_ITER_REACHED
};
-int EMD_wrap(int n1,int n2, double *X, double *Y,double *D, double *G, double *cost, int max_iter);
+int EMD_wrap(int n1,int n2, double *X, double *Y,double *D, double *G, double* alpha, double* beta, double *cost, int maxIter);
#endif