diff options
Diffstat (limited to 'examples/backends/plot_stoch_continuous_ot_pytorch.py')
| -rw-r--r-- | examples/backends/plot_stoch_continuous_ot_pytorch.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/examples/backends/plot_stoch_continuous_ot_pytorch.py b/examples/backends/plot_stoch_continuous_ot_pytorch.py index 6d9b916..714a5d3 100644 --- a/examples/backends/plot_stoch_continuous_ot_pytorch.py +++ b/examples/backends/plot_stoch_continuous_ot_pytorch.py @@ -125,8 +125,8 @@ pl.xlabel("Iterations") # %% -# Plot the density on arget for a given source sample -# --------------------------------------------------- +# Plot the density on target for a given source sample +# ---------------------------------------------------- nv = 100 @@ -155,7 +155,7 @@ Gg = Gg.reshape((nv, nv)).detach().numpy() pl.scatter(Xs[:nvisu, 0], Xs[:nvisu, 1], marker='+', zorder=2, alpha=0.05) pl.scatter(Xt[:nvisu, 0], Xt[:nvisu, 1], marker='o', zorder=2, alpha=0.05) pl.scatter(Xs[iv:iv + 1, 0], Xs[iv:iv + 1, 1], s=100, marker='+', label='Source sample', zorder=2, alpha=1, color='C0') -pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported sourec sample') +pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported source sample') pl.legend(loc=0) ax_bounds = pl.axis() pl.title('Density of transported source sample') @@ -169,7 +169,7 @@ Gg = Gg.reshape((nv, nv)).detach().numpy() pl.scatter(Xs[:nvisu, 0], Xs[:nvisu, 1], marker='+', zorder=2, alpha=0.05) pl.scatter(Xt[:nvisu, 0], Xt[:nvisu, 1], marker='o', zorder=2, alpha=0.05) pl.scatter(Xs[iv:iv + 1, 0], Xs[iv:iv + 1, 1], s=100, marker='+', label='Source sample', zorder=2, alpha=1, color='C0') -pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported sourec sample') +pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported source sample') pl.legend(loc=0) ax_bounds = pl.axis() pl.title('Density of transported source sample') @@ -183,7 +183,7 @@ Gg = Gg.reshape((nv, nv)).detach().numpy() pl.scatter(Xs[:nvisu, 0], Xs[:nvisu, 1], marker='+', zorder=2, alpha=0.05) pl.scatter(Xt[:nvisu, 0], Xt[:nvisu, 1], marker='o', zorder=2, alpha=0.05) pl.scatter(Xs[iv:iv + 1, 0], Xs[iv:iv + 1, 1], s=100, marker='+', label='Source sample', zorder=2, alpha=1, color='C0') -pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported sourec sample') +pl.pcolormesh(XX, YY, Gg, cmap='Greens', label='Density of transported source sample') pl.legend(loc=0) ax_bounds = pl.axis() pl.title('Density of transported source sample') |