Merge pull request #84 from alexisthual/feat/stabilized-sinkhorn

[ENH] Stabilized and epsilon scaled sinkhorn solvers

Author: pbarbarant

src/fugw/solvers/dense.py

@@ -11,7 +11,7 @@
     compute_divergence,
     solver_ibpp,
     solver_mm,
-    solver_sinkhorn,
+    solver_sinkhorn_log,
     solver_mm_l2,
 )
 from fugw.utils import _add_dict, console
@@ -444,7 +444,7 @@ def solve(
 
         # If divergence is KL
         self_solver_sinkhorn = partial(
-            solver_sinkhorn,
+            solver_sinkhorn_log,
             tuple_weights=(ws, wt, ws_dot_wt),
             train_params=(self.nits_uot, self.tol_uot, self.eval_uot),
         )

src/fugw/solvers/sparse.py

@@ -17,7 +17,7 @@
     crow_indices_to_row_indices,
     csr_sum,
     elementwise_prod_fact_sparse,
-    solver_sinkhorn_sparse,
+    solver_sinkhorn_log_sparse,
     solver_ibpp_sparse,
     solver_mm_sparse,
     solver_mm_l2_sparse,
@@ -517,7 +517,7 @@ def solve(
 
         # If divergence is KL
         self_solver_sinkhorn = partial(
-            solver_sinkhorn_sparse,
+            solver_sinkhorn_log_sparse,
             tuple_weights=(ws, wt, ws_dot_wt),
             train_params=(self.nits_uot, self.tol_uot, self.eval_uot),
             verbose=verbose,