Update results.py

results.py

@@ -6,18 +6,19 @@ from fh_matplotlib import matplotlib2fasthtml
 
 
 with open("data/graph_data.jsonl", "r") as f:
-    for line in f:
-        for title, meta in json.loads(line).items():
-            _, ax = pyplot.subplots()
-            for i, (label, data, color) in enumerate(meta["data"]):
-                ax.plot(*data, label=label, color=color)
-
-            ax.grid()
-            ax.set_xlabel(meta["x_label"])
-            ax.set_ylabel(meta["y_label"])
-            ax.set_title(title)
-            ax.legend()
-
+    if title = "Perplexity Across Different Buckets (global)":
+        for line in f:
+            for title, meta in json.loads(line).items():
+                _, ax = pyplot.subplots()
+                for i, (label, data, color) in enumerate(meta["data"]):
+                    ax.plot(*data, label=label, color=color)
+    
+                ax.grid()
+                ax.set_xlabel(meta["x_label"])
+                ax.set_ylabel(meta["y_label"])
+                ax.set_title(title)
+                ax.legend()
+        
 fig = pyplot.show()
 
 intro_div = Div(