diff options
Diffstat (limited to 'examples/sliced-wasserstein/plot_variance_ssw.py')
| -rw-r--r-- | examples/sliced-wasserstein/plot_variance_ssw.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/examples/sliced-wasserstein/plot_variance_ssw.py b/examples/sliced-wasserstein/plot_variance_ssw.py index f5fc35f..246b2a8 100644 --- a/examples/sliced-wasserstein/plot_variance_ssw.py +++ b/examples/sliced-wasserstein/plot_variance_ssw.py @@ -28,7 +28,7 @@ import ot # %% parameters and data generation -n = 500 # nb samples +n = 200 # nb samples xs = np.random.randn(n, 3) xt = np.random.randn(n, 3) @@ -81,9 +81,9 @@ pl.title("Source and Target distribution") # Spherical Sliced Wasserstein for different seeds and number of projections # -------------------------------------------------------------------------- -n_seed = 50 -n_projections_arr = np.logspace(0, 3, 25, dtype=int) -res = np.empty((n_seed, 25)) +n_seed = 20 +n_projections_arr = np.logspace(0, 3, 10, dtype=int) +res = np.empty((n_seed, 10)) # %% Compute statistics for seed in range(n_seed): |