update plotting code

[ml_measurement_error_public.git] / simulations / grid_sweep.py

diff --git a/simulations/grid_sweep.py b/simulations/grid_sweep.py
index 86312ea77c6cf572e142cdaf846738c74f3b529d..b4287299938bb4ecae82db12dc830bc1e340809c 100755 (executable)
--- a/simulations/grid_sweep.py
+++ b/simulations/grid_sweep.py
@@ -2,8 +2,14 @@
 
 import fire
 from itertools import product
+import pyRemembeR
 
-def main(command, arg_dict, outfile):
+def main(command, arg_dict, outfile, remember_file='remember_grid_sweep.RDS'):
+    print(remember_file)
+    remember = pyRemembeR.remember.Remember()
+    remember.set_file(remember_file)
+    remember[outfile] = arg_dict
+    remember.save_to_r()
     keys = []
     values = []