Skip to content
Snippets Groups Projects
Unverified Commit c4a9fe53 authored by kossi's avatar kossi Committed by GitHub
Browse files

Merge pull request #1 from kossi-kossivi/kossi-kossivi-patch-1

Update dataset.py
parents 2e5930aa e8a219c5
Branches
Tags
No related merge requests found
......@@ -106,9 +106,9 @@ class Dataset():
return concat_views, view_limits
def select_labels(self, selected_label_names):
selected_labels = [self.get_label_names().index(label_name.decode())
selected_labels = [self.get_label_names(decode=True).index(label_name.decode())
if isinstance(label_name, bytes)
else self.get_label_names().index(label_name)
else self.get_label_names(decode=True).index(label_name)
for label_name in selected_label_names]
selected_indices = np.array([index
for index, label in
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment