Commit ec7f1e17 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

make sure the biofiles are kept

parent f83a7087
......@@ -57,7 +57,7 @@ class BioGenerator(object):
return -1
self.database = database
self.biofiles = biofiles
self.biofiles = list(biofiles)
self.load_data = load_data
self.biofile_to_label = biofile_to_label
self.multiple_samples = multiple_samples
......@@ -77,9 +77,9 @@ class BioGenerator(object):
self._output_shapes = (
data.shape, tf.TensorShape([]), tf.TensorShape([]))
logger.debug("Initializing a dataset with %d files and %s types "
"and %s shapes", len(self.biofiles), self.output_types,
self.output_shapes)
logger.info("Initializing a dataset with %d files and %s types "
"and %s shapes", len(self.biofiles), self.output_types,
self.output_shapes)
@property
def labels(self):
......
Markdown is supported
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