Merged
Merged
Conversation
Kilian Fatras added 5 commits
March 29, 2019 12:41
rflamary
changed the title
add empirical sinkhorn and sikhorn divergence functions
add empirical sinkhorn and sinkhorn divergence functions
Kilian Fatras and others added 2 commits
March 31, 2019 12:14
| .. [23] Aude Genevay, Gabriel Peyré, Marco Cuturi, Learning Generative Models with Sinkhorn Divergences, Proceedings of the Twenty-First International Conference on Artficial Intelligence and Statistics, (AISTATS) 21, 2018 | ||
| ''' | ||
|
|
||
| sinkhorn_div = (2 * empirical_sinkhorn2(X_s, X_t, reg, a, b, metric=metric, numIterMax=numIterMax, stopThr=1e-9, verbose=verbose, log=log, **kwargs) - |
| M_t = ot.dist(X_t, X_t) | ||
|
|
||
| emp_sinkhorn_div = ot.bregman.empirical_sinkhorn_divergence(X_s, X_t, 1) | ||
| sinkhorn_div = (2 * ot.sinkhorn2(a, b, M, 1) - ot.sinkhorn2(a, a, M_s, 1) - |
Kilian Fatras added 4 commits
April 4, 2019 13:45
Labels
None yet