brain maps

Author: lucidtronix

ml4h/recipes.py

@@ -1372,8 +1372,8 @@ def _scalar_predictions_from_generator(args, models_inputs_outputs, generator, s
     predictions = defaultdict(dict)
     for j in range(steps):
         batch = next(iter(generator))
-        input_data, output_data, tensor_paths = batch[BATCH_INPUT_INDEX], batch[BATCH_OUTPUT_INDEX], batch[BATCH_PATHS_INDEX]
-        test_paths.extend(tensor_paths)
+        input_data, output_data = batch[BATCH_INPUT_INDEX], batch[BATCH_OUTPUT_INDEX] #, batch[BATCH_PATHS_INDEX]
+        # test_paths.extend(tensor_paths)
         for tl in test_labels:
             test_labels[tl].extend(np.copy(output_data[tl]))