From 112fff2edd08c32d9927d858e5e73659eab96ab6 Mon Sep 17 00:00:00 2001
From: Baptiste Bauvin <baptiste.bauvin@lis-lab.fr>
Date: Mon, 25 Nov 2019 14:02:57 -0500
Subject: [PATCH] Labels selection
---
multiview_platform/mono_multi_view_classifiers/utils/dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/multiview_platform/mono_multi_view_classifiers/utils/dataset.py b/multiview_platform/mono_multi_view_classifiers/utils/dataset.py
index 85666b66..e7692142 100644
--- a/multiview_platform/mono_multi_view_classifiers/utils/dataset.py
+++ b/multiview_platform/mono_multi_view_classifiers/utils/dataset.py
@@ -292,7 +292,7 @@ class Dataset():
self.dataset.copy("Metadata", new_dataset_file)
if "example_ids" in self.dataset["Metadata"].keys():
ex_ids = new_dataset_file["Metadata"]["example_ids"]
- ex_ids = np.array([self.example_ids[example_indices]]).astype(np.dtype("S10"))
+ ex_ids[...] = np.array(self.example_ids)[example_indices].astype(np.dtype("S10"))
else:
new_dataset_file["Metadata"].create_dataset("example_ids",
(len(self.example_ids), ),
--
GitLab