Skip to content
Snippets Groups Projects
Commit f83a7087 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Keep regenerating labels and keys

parent 93086e0e
No related branches found
No related tags found
1 merge request!33Changes to the biogenerator
...@@ -56,8 +56,6 @@ class BioGenerator(object): ...@@ -56,8 +56,6 @@ class BioGenerator(object):
def biofile_to_label(biofile): def biofile_to_label(biofile):
return -1 return -1
self.labels = (biofile_to_label(f) for f in biofiles)
self.keys = (str(f.make_path("", "")) for f in biofiles)
self.database = database self.database = database
self.biofiles = biofiles self.biofiles = biofiles
self.load_data = load_data self.load_data = load_data
...@@ -83,6 +81,16 @@ class BioGenerator(object): ...@@ -83,6 +81,16 @@ class BioGenerator(object):
"and %s shapes", len(self.biofiles), self.output_types, "and %s shapes", len(self.biofiles), self.output_types,
self.output_shapes) self.output_shapes)
@property
def labels(self):
for f in self.biofiles:
yield self.biofile_to_label(f)
@property
def keys(self):
for f in self.biofiles:
yield str(f.make_path("", "")).encode('utf-8')
@property @property
def output_types(self): def output_types(self):
return self._output_types return self._output_types
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment