diff options
Diffstat (limited to 'examples')
-rw-r--r-- | examples/domain-adaptation/plot_otda_linear_mapping.py | 2 | ||||
-rw-r--r-- | examples/gromov/plot_barycenter_fgw.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/examples/domain-adaptation/plot_otda_linear_mapping.py b/examples/domain-adaptation/plot_otda_linear_mapping.py index a44096a..8284a2a 100644 --- a/examples/domain-adaptation/plot_otda_linear_mapping.py +++ b/examples/domain-adaptation/plot_otda_linear_mapping.py @@ -61,7 +61,7 @@ plt.plot(xt[:, 0], xt[:, 1], 'o') # Estimate linear mapping and transport # ------------------------------------- -Ae, be = ot.da.OT_mapping_linear(xs, xt) +Ae, be = ot.gaussian.empirical_bures_wasserstein_mapping(xs, xt) xst = xs.dot(Ae) + be diff --git a/examples/gromov/plot_barycenter_fgw.py b/examples/gromov/plot_barycenter_fgw.py index 556e08f..dc3c6aa 100644 --- a/examples/gromov/plot_barycenter_fgw.py +++ b/examples/gromov/plot_barycenter_fgw.py @@ -174,7 +174,7 @@ A, C, log = fgw_barycenters(sizebary, Ys, Cs, ps, lambdas, alpha=0.95, log=True) # ------------------------- #%% Create the barycenter -bary = nx.from_numpy_matrix(sp_to_adjency(C, threshinf=0, threshsup=find_thresh(C, sup=100, step=100)[0])) +bary = nx.from_numpy_array(sp_to_adjency(C, threshinf=0, threshsup=find_thresh(C, sup=100, step=100)[0])) for i, v in enumerate(A.ravel()): bary.add_node(i, attr_name=v) |