(samples_source, samples_target)
| 121 | |
| 122 | |
| 123 | def benchmark_naive(samples_source, samples_target): |
| 124 | start = perf_counter() |
| 125 | for i in range(n_problems): |
| 126 | M = ot.dist(samples_source[i], samples_target[i]) |
| 127 | res = ot.solve(M, reg=reg, max_iter=max_iter, tol=tol, reg_type="entropy") |
| 128 | end = perf_counter() |
| 129 | return end - start |
| 130 | |
| 131 | |
| 132 | def benchmark_batch(samples_source, samples_target): |