diff options
Diffstat (limited to 'test/test_da.py')
-rw-r--r-- | test/test_da.py | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/test/test_da.py b/test/test_da.py index 8df4795..a38390f 100644 --- a/test/test_da.py +++ b/test/test_da.py @@ -3,18 +3,15 @@ import numpy as np import ot -# import pytest - - def test_otda(): - n = 150 # nb samples + n_samples = 150 # nb samples np.random.seed(0) - xs, ys = ot.datasets.get_data_classif('3gauss', n) - xt, yt = ot.datasets.get_data_classif('3gauss2', n) + xs, ys = ot.datasets.get_data_classif('3gauss', n_samples) + xt, yt = ot.datasets.get_data_classif('3gauss2', n_samples) - a, b = ot.unif(n), ot.unif(n) + a, b = ot.unif(n_samples), ot.unif(n_samples) # LP problem da_emd = ot.da.OTDA() # init class |