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

make sure the biofiles are kept

parent f83a7087
Branches
No related tags found
1 merge request!33Changes to the biogenerator
...@@ -57,7 +57,7 @@ class BioGenerator(object): ...@@ -57,7 +57,7 @@ class BioGenerator(object):
return -1 return -1
self.database = database self.database = database
self.biofiles = biofiles self.biofiles = list(biofiles)
self.load_data = load_data self.load_data = load_data
self.biofile_to_label = biofile_to_label self.biofile_to_label = biofile_to_label
self.multiple_samples = multiple_samples self.multiple_samples = multiple_samples
...@@ -77,9 +77,9 @@ class BioGenerator(object): ...@@ -77,9 +77,9 @@ class BioGenerator(object):
self._output_shapes = ( self._output_shapes = (
data.shape, tf.TensorShape([]), tf.TensorShape([])) data.shape, tf.TensorShape([]), tf.TensorShape([]))
logger.debug("Initializing a dataset with %d files and %s types " logger.info("Initializing a dataset with %d files and %s types "
"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 @property
def labels(self): def labels(self):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment