Commit f83a7087 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

Keep regenerating labels and keys

parent 93086e0e
......@@ -56,8 +56,6 @@ class BioGenerator(object):
def biofile_to_label(biofile):
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.biofiles = biofiles
self.load_data = load_data
......@@ -83,6 +81,16 @@ class BioGenerator(object):
"and %s shapes", len(self.biofiles), self.output_types,
def labels(self):
for f in self.biofiles:
yield self.biofile_to_label(f)
def keys(self):
for f in self.biofiles:
yield str(f.make_path("", "")).encode('utf-8')
def output_types(self):
return self._output_types
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment