diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2017-09-15 14:54:21 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2017-09-15 14:54:21 +0200 |
commit | 81b2796226f3abde29fc024752728444da77509a (patch) | |
tree | c52cec3c38552f9f8c15361758aa9a80c30c3ef3 /docs/source/auto_examples/demo_OT_2D_sampleslarge.rst | |
parent | e70d5420204db78691af2d0fbe04cc3d4416a8f4 (diff) | |
parent | 7fea2cd3e8ad29bf3fa442d7642bae124ee2bab0 (diff) |
Merge pull request #27 from rflamary/autonb
auto notebooks + release update (fixes #16)
Diffstat (limited to 'docs/source/auto_examples/demo_OT_2D_sampleslarge.rst')
-rw-r--r-- | docs/source/auto_examples/demo_OT_2D_sampleslarge.rst | 106 |
1 files changed, 0 insertions, 106 deletions
diff --git a/docs/source/auto_examples/demo_OT_2D_sampleslarge.rst b/docs/source/auto_examples/demo_OT_2D_sampleslarge.rst deleted file mode 100644 index f5dbb0d..0000000 --- a/docs/source/auto_examples/demo_OT_2D_sampleslarge.rst +++ /dev/null @@ -1,106 +0,0 @@ - - -.. _sphx_glr_auto_examples_demo_OT_2D_sampleslarge.py: - - -Demo for 2D Optimal transport between empirical distributions - -@author: rflamary - - - -.. code-block:: python - - - import numpy as np - import matplotlib.pylab as pl - import ot - - #%% parameters and data generation - - n=5000 # nb samples - - mu_s=np.array([0,0]) - cov_s=np.array([[1,0],[0,1]]) - - mu_t=np.array([4,4]) - cov_t=np.array([[1,-.8],[-.8,1]]) - - xs=ot.datasets.get_2D_samples_gauss(n,mu_s,cov_s) - xt=ot.datasets.get_2D_samples_gauss(n,mu_t,cov_t) - - a,b = ot.unif(n),ot.unif(n) # uniform distribution on samples - - # loss matrix - M=ot.dist(xs,xt) - M/=M.max() - - #%% plot samples - - #pl.figure(1) - #pl.plot(xs[:,0],xs[:,1],'+b',label='Source samples') - #pl.plot(xt[:,0],xt[:,1],'xr',label='Target samples') - #pl.legend(loc=0) - #pl.title('Source and traget distributions') - # - #pl.figure(2) - #pl.imshow(M,interpolation='nearest') - #pl.title('Cost matrix M') - # - - #%% EMD - - G0=ot.emd(a,b,M) - - #pl.figure(3) - #pl.imshow(G0,interpolation='nearest') - #pl.title('OT matrix G0') - # - #pl.figure(4) - #ot.plot.plot2D_samples_mat(xs,xt,G0,c=[.5,.5,1]) - #pl.plot(xs[:,0],xs[:,1],'+b',label='Source samples') - #pl.plot(xt[:,0],xt[:,1],'xr',label='Target samples') - #pl.legend(loc=0) - #pl.title('OT matrix with samples') - - - #%% sinkhorn - - # reg term - lambd=5e-3 - - Gs=ot.sinkhorn(a,b,M,lambd) - - #pl.figure(5) - #pl.imshow(Gs,interpolation='nearest') - #pl.title('OT matrix sinkhorn') - # - #pl.figure(6) - #ot.plot.plot2D_samples_mat(xs,xt,Gs,color=[.5,.5,1]) - #pl.plot(xs[:,0],xs[:,1],'+b',label='Source samples') - #pl.plot(xt[:,0],xt[:,1],'xr',label='Target samples') - #pl.legend(loc=0) - #pl.title('OT matrix Sinkhorn with samples') - # - - -**Total running time of the script:** ( 0 minutes 0.000 seconds) - - - -.. container:: sphx-glr-footer - - - .. container:: sphx-glr-download - - :download:`Download Python source code: demo_OT_2D_sampleslarge.py <demo_OT_2D_sampleslarge.py>` - - - - .. container:: sphx-glr-download - - :download:`Download Jupyter notebook: demo_OT_2D_sampleslarge.ipynb <demo_OT_2D_sampleslarge.ipynb>` - -.. rst-class:: sphx-glr-signature - - `Generated by Sphinx-Gallery <http://sphinx-gallery.readthedocs.io>`_ |