Skip to content
Snippets Groups Projects

Use bob.extension to load config files

Merged Amir MOHAMMADI requested to merge predict into master
8 files
+ 17
14
Compare changes
  • Side-by-side
  • Inline
Files
8
@@ -41,7 +41,8 @@ class BioGenerator(object):
"""
def __init__(self, database, biofiles, load_data=None,
biofile_to_label=None, multiple_samples=False):
biofile_to_label=None, multiple_samples=False, **kwargs):
super(BioGenerator, self).__init__(**kwargs)
if load_data is None:
def load_data(database, biofile):
data = read_original_data(
@@ -95,6 +96,9 @@ class BioGenerator(object):
def output_shapes(self):
return self._output_shapes
def __len__(self):
return len(self.biofiles)
def __call__(self):
"""A generator function that when called will return the samples.
Loading