victormiller commited on
Commit
b87e744
1 Parent(s): 93e2035

Update results.py

Browse files
Files changed (1) hide show
  1. results.py +12 -12
results.py CHANGED
@@ -6,18 +6,18 @@ from fh_matplotlib import matplotlib2fasthtml
6
 
7
 
8
  with open("data/graph_data.jsonl", "r") as f:
9
- if title == "Perplexity Across Different Buckets (global)":
10
- for line in f:
11
- for title, meta in json.loads(line).items():
12
- _, ax = pyplot.subplots()
13
- for i, (label, data, color) in enumerate(meta["data"]):
14
- ax.plot(*data, label=label, color=color)
15
-
16
- ax.grid()
17
- ax.set_xlabel(meta["x_label"])
18
- ax.set_ylabel(meta["y_label"])
19
- ax.set_title(title)
20
- ax.legend()
21
 
22
  fig = pyplot.show()
23
 
 
6
 
7
 
8
  with open("data/graph_data.jsonl", "r") as f:
9
+ #if title == "Perplexity Across Different Buckets (global)":
10
+ for line in f:
11
+ for title, meta in json.loads(line).items():
12
+ _, ax = pyplot.subplots()
13
+ for i, (label, data, color) in enumerate(meta["data"]):
14
+ ax.plot(*data, label=label, color=color)
15
+
16
+ ax.grid()
17
+ ax.set_xlabel(meta["x_label"])
18
+ ax.set_ylabel(meta["y_label"])
19
+ ax.set_title(title)
20
+ ax.legend()
21
 
22
  fig = pyplot.show()
23