improve logging of predict_bio

bob/learn/tensorflow/dataset/bio.py

@@ -42,6 +42,7 @@ class BioGenerator(object):
 
     def __init__(self, database, biofiles, load_data=None,
                  biofile_to_label=None, multiple_samples=False):
+        super(self, BioGenerator).__init__()
         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.
 

bob/learn/tensorflow/script/predict_bio.py

@@ -231,7 +231,8 @@ def main(argv=None):
         checkpoint_path=checkpoint_path,
     )
 
-    logger.info("Saving the predictions in %s", output_dir)
+    logger.info("Saving the predictions of %d files in %s", len(generator),
+                output_dir)
 
     pool = Pool()
     try: