Skip to content
Snippets Groups Projects

Changes to the biogenerator

Merged Amir MOHAMMADI requested to merge predict into master
All threads resolved!
1 file
+ 4
4
Compare changes
  • Side-by-side
  • Inline
@@ -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):
Loading