blob: af5269c69b3c8d6a1d10bd71d13a2100244607f3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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)
|
