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 | |
parent | 449045fda6070c3b9b5a599305357713a164e7ee (diff) |
small update
-rw-r--r-- | examples/plot_OTDA_classes.py | 1 | ||||
-rw-r--r-- | ot/datasets.py | 6 |
2 files changed, 4 insertions, 3 deletions
diff --git a/examples/plot_OTDA_classes.py b/examples/plot_OTDA_classes.py index 999be53..c00cef6 100644 --- a/examples/plot_OTDA_classes.py +++ b/examples/plot_OTDA_classes.py @@ -11,6 +11,7 @@ import ot + #%% parameters n=150 # nb samples in source and target datasets 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 |