diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2017-06-13 15:50:11 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-06-13 15:50:11 +0200 |
| commit | 38e96f88eb520b9fa8333686565b082d2921e131 (patch) | |
| tree | a0644f3fe062e9f6b8ba4c279934b0311508d78c /examples/plot_compute_emd.py | |
| parent | a7bed093f91922e18fa5902c4d1d63b9712d5794 (diff) | |
implement paralell sinkhorn stabilized
Diffstat (limited to 'examples/plot_compute_emd.py')
| -rw-r--r-- | examples/plot_compute_emd.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/examples/plot_compute_emd.py b/examples/plot_compute_emd.py index 226bc97..08de6ee 100644 --- a/examples/plot_compute_emd.py +++ b/examples/plot_compute_emd.py @@ -16,7 +16,7 @@ from ot.datasets import get_1D_gauss as gauss #%% parameters n=100 # nb bins -n_target=10 # nb target distributions +n_target=50 # nb target distributions # bin positions @@ -61,14 +61,15 @@ pl.legend() #%% reg=1e-2 -d_sinkhorn=ot.sinkhorn(a,B,M,reg) +d_sinkhorn=ot.sinkhorn(a,B,M,reg,method='sinkhorn_stabilized') +d_sinkhorn0=ot.sinkhorn(a,B,M,reg) d_sinkhorn2=ot.sinkhorn(a,B,M2,reg) pl.figure(2) pl.clf() pl.plot(d_emd,label='Euclidean EMD') pl.plot(d_emd2,label='Squared Euclidean EMD') -pl.plot(d_sinkhorn,label='Euclidean Sinkhorn') -pl.plot(d_emd2,label='Squared Euclidean Sinkhorn') +pl.plot(d_sinkhorn,'+',label='Euclidean Sinkhorn') +pl.plot(d_sinkhorn2,'+',label='Squared Euclidean Sinkhorn') pl.title('EMD distances') pl.legend()
\ No newline at end of file |