save results to json file

experiment1.py

@@ -1,5 +1,6 @@
 import argparse
 import json
+import datetime
 from multiprocessing import Pool
 from pathlib import Path
 
@@ -53,6 +54,9 @@ def gen_args(config_fn):
     if args.test:
         sim_args = sim_args[:2]
 
+    # date of simulation in string format
+    date_str = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
+
     all_results = []
 
     if args.seq:
@@ -64,3 +68,8 @@ def gen_args(config_fn):
             results = p.map(process, sim_args)
             for r in results:
                 all_results += r
+
+    filename = f"{date_str}_smd_results.json"
+    with open(filename, "w") as f:
+        json.dump(all_results, f)
+

process.py

@@ -125,9 +125,11 @@ def process(args):
                 Y, X[:, :, ref_mic], p=p, q=q, **kwargs
             )
 
-            y, t["sdr"], t["sir"], _ = reconstruct_evaluate(
+            y, sdr, sir, _ = reconstruct_evaluate(
                 ref, Z, nfft, hop, win=win_s
             )
+            t["sdr"] = sdr.tolist()
+            t["sir"] = sir.tolist()
             results.append(t.copy())
 
     return results