noise repair, fingerprint check

2020-05-15 19:23:05 +02:00
parent 4a6b32d1dd
commit d88a6bcf71
3 changed files with 40 additions and 28 deletions
@@ -12,6 +12,8 @@ config_file_name = 'config.ini'

 if __name__ == '__main__':
    for model_path in outpath.iterdir():
+        if not model_path.is_dir():
+            continue
        out_file = (model_path / metric_file_name)
        for paramter_configuration in model_path.iterdir():
            uar_scores = defaultdict(list)
@@ -46,5 +48,8 @@ if __name__ == '__main__':
                writer = csv.DictWriter(f, delimiter=',', lineterminator='\n', fieldnames=headers)
                if not file_existed:
                    writer.writeheader()  # file doesn't exist yet, write a header
-                for row_idx in range(len(uar_scores['mean'])):
-                    writer.writerow({key: uar_scores[key][row_idx] for key in headers})
+                try:
+                    for row_idx in range(len(uar_scores['mean'])):
+                        writer.writerow({key: uar_scores[key][row_idx] for key in headers})
+                except IndexError:
+                    print('could not read: ')