diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2016-12-02 14:29:22 +0100 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2016-12-02 14:29:22 +0100 |
commit | 01f15d4ac4a909ca7e71f6b8f6bca3edf88d7e47 (patch) | |
tree | a838ccedf1eac41ff549840a5bd6e401261b2dc9 /ot | |
parent | 449045fda6070c3b9b5a599305357713a164e7ee (diff) |
small update
Diffstat (limited to 'ot')
-rw-r--r-- | ot/datasets.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/ot/datasets.py b/ot/datasets.py index 5c1ef78..7816833 100644 --- a/ot/datasets.py +++ b/ot/datasets.py @@ -124,8 +124,8 @@ def get_data_classif(dataset,n,nz=.5,theta=0,**kwargs): else: - x=0 - y=0 + x=np.array(0) + y=np.array(0) print("unknown dataset") - return x,y
\ No newline at end of file + return x,y.astype(int)
\ No newline at end of file |