narrative matching: support for formatting structural blocks results

CompNet · Mar 10, 2024 · 97e1619 · 97e1619
1 parent 16abe9c
commit 97e1619
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/matching/plot/format_structural_results.py b/src/matching/plot/format_structural_results.py
@@ -22,6 +22,9 @@
         action="store_true",
         help="if specified, use cumulative results",
     )
+    parser.add_argument(
+        "-b", "--blocks", action="store_true"
+    )
     args = parser.parse_args()
 
     dfs_dict = {}
@@ -31,6 +34,8 @@
             path = (
                 f"{root_dir}/out/matching/plot/{medias}_structural_cumulative/df.pickle"
             )
+        if args.blocks:
+            path = f"{root_dir}/out/matching/plot/{medias}_structural_blocks/df.pickle"
         with open(path, "rb") as f:
             df = pickle.load(f)
             df = df.loc[