-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathplot_results.py
More file actions
50 lines (40 loc) · 1.43 KB
/
plot_results.py
File metadata and controls
50 lines (40 loc) · 1.43 KB
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import csv
import os
import matplotlib.pyplot as plt
RESULTS_FILE = "results/benchmark_results.csv"
def main():
os.makedirs("results", exist_ok=True)
task_counts = []
naive_times = []
optimized_times = []
speedups = []
with open(RESULTS_FILE, "r") as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
task_counts.append(int(row["task_count"]))
naive_times.append(float(row["naive_time"]))
optimized_times.append(float(row["optimized_time"]))
speedups.append(float(row["speedup"]))
plt.figure()
plt.plot(task_counts, naive_times, marker="o", label="Naive")
plt.plot(task_counts, optimized_times, marker="o", label="Optimized")
plt.xlabel("Task Count")
plt.ylabel("Processing Time (s)")
plt.title("Naive vs Optimized Processing Time")
plt.legend()
plt.tight_layout()
plt.savefig("results/processing_time_comparison.png")
plt.close()
plt.figure()
plt.plot(task_counts, speedups, marker="o")
plt.xlabel("Task Count")
plt.ylabel("Speedup (x)")
plt.title("Optimized Engine Speedup by Task Count")
plt.tight_layout()
plt.savefig("results/speedup_by_task_count.png")
plt.close()
print("Saved plots:")
print("- results/processing_time_comparison.png")
print("- results/speedup_by_task_count.png")
if __name__ == "__main__":
main()