1 files changed, 26 insertions, 0 deletions
diff --git a/simulation/plot_utils.py b/simulation/plot_utils.py
new file mode 100644
index 0000000..af5269c
--- /dev/null
+++ b/simulation/plot_utils.py
@@ -0,0 +1,26 @@
+import matplotlib.pyplot as plt
+import argparse
+import json
+import seaborn
+seaborn.set_style('whitegrid')
+
+parser = argparse.ArgumentParser(description='Process logs')
+parser.add_argument('-x', help='name of parameters on x axis', default='time')
+parser.add_argument('-y', help='name of parameters on y axis', default='rmse')
+parser.add_argument('f', help='list of logs to parse', nargs='+')
+parser.add_argument('-dest', help='name of figure to save', default='fig.png')
+args = parser.parse_args()
+
+for file_name in args.f:
+    x, y = [], []
+    with open(file_name) as f:
+        for line in f:
+            jason = json.loads(line)
+            x.append(jason[args.x])
+            y.append(jason[args.y])
+    plt.plot(x, y, label=file_name)
+
+plt.legend()
+plt.xlabel(args.x)
+plt.ylabel(args.y)
+plt.savefig(args.dest)