diff options
| author | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2017-07-12 23:56:27 +0200 |
|---|---|---|
| committer | Alexandre Gramfort <alexandre.gramfort@m4x.org> | 2017-07-20 14:08:02 +0200 |
| commit | 95db977e8b931277af5dadbd79eccbd5fbb8bb62 (patch) | |
| tree | 4b48e558afceddfcb640e42bb0682bdcccb02b55 /test/test_gpu_sinkhorn.py | |
| parent | 37ca3142a4c8c808382f5eb1c23bf198c3e4610e (diff) | |
pep8
Diffstat (limited to 'test/test_gpu_sinkhorn.py')
| -rw-r--r-- | test/test_gpu_sinkhorn.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_gpu_sinkhorn.py b/test/test_gpu_sinkhorn.py index bfa2cd2..841f062 100644 --- a/test/test_gpu_sinkhorn.py +++ b/test/test_gpu_sinkhorn.py @@ -3,8 +3,10 @@ import numpy as np import time import ot.gpu + def describeRes(r): - print("min:{:.3E}, max::{:.3E}, mean::{:.3E}, std::{:.3E}".format(np.min(r),np.max(r),np.mean(r),np.std(r))) + print("min:{:.3E}, max::{:.3E}, mean::{:.3E}, std::{:.3E}".format( + np.min(r), np.max(r), np.mean(r), np.std(r))) for n in [5000, 10000, 15000, 20000]: @@ -23,4 +25,4 @@ for n in [5000, 10000, 15000, 20000]: print("Normal sinkhorn, time: {:6.2f} sec ".format(time2 - time1)) describeRes(G1) print(" GPU sinkhorn, time: {:6.2f} sec ".format(time3 - time2)) - describeRes(G2)
\ No newline at end of file + describeRes(G2) |