add fgw examples in doc

1 files changed, 55 insertions, 1 deletions

diff --git a/docs/source/auto_examples/plot_OT_2D_samples.rst b/docs/source/auto_examples/plot_OT_2D_samples.rst
index 624ae3e..1f1d713 100644
--- a/docs/source/auto_examples/plot_OT_2D_samples.rst
+++ b/docs/source/auto_examples/plot_OT_2D_samples.rst
@@ -17,6 +17,7 @@ sum of diracs. The OT matrix is plotted with the samples.
 
 
     # Author: Remi Flamary <remi.flamary@unice.fr>
+    #         Kilian Fatras <kilian.fatras@irisa.fr>
     #
     # License: MIT License
 
@@ -176,6 +177,8 @@ Compute Sinkhorn
 
 
 
+
+
 .. rst-class:: sphx-glr-horizontal
 
 
@@ -192,7 +195,58 @@ Compute Sinkhorn
 
 
 
-**Total running time of the script:** ( 0 minutes  3.027 seconds)
+Emprirical Sinkhorn
+----------------
+
+
+
+.. code-block:: python
+
+
+    #%% sinkhorn
+
+    # reg term
+    lambd = 1e-3
+
+    Ges = ot.bregman.empirical_sinkhorn(xs, xt, lambd)
+
+    pl.figure(7)
+    pl.imshow(Ges, interpolation='nearest')
+    pl.title('OT matrix empirical sinkhorn')
+
+    pl.figure(8)
+    ot.plot.plot2D_samples_mat(xs, xt, Ges, 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 from samples')
+
+    pl.show()
+
+
+
+.. rst-class:: sphx-glr-horizontal
+
+
+    *
+
+      .. image:: /auto_examples/images/sphx_glr_plot_OT_2D_samples_013.png
+            :scale: 47
+
+    *
+
+      .. image:: /auto_examples/images/sphx_glr_plot_OT_2D_samples_014.png
+            :scale: 47
+
+
+.. rst-class:: sphx-glr-script-out
+
+ Out::
+
+    Warning: numerical errors at iteration 0
+
+
+**Total running time of the script:** ( 0 minutes  2.616 seconds)