sharkdp/hyperfine
1
1
Fork 0
mirror of https://github.com/sharkdp/hyperfine.git synced 2024-09-11 14:07:02 +03:00
Code Issues Projects Releases Wiki Activity

Improve plot_progression.py script

Browse Source
This commit is contained in:
David Peter 2024-02-02 17:51:31 +01:00 committed by David Peter
parent a75dca90e1
commit c7be935da3
1 changed files with 29 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
scripts/plot_progression.py
View File

@ -30,32 +30,48 @@ parser.add_argument(
metavar="num_runs",
help="Width of the moving-average window (default: N/5)",
)
parser.add_argument(
"--no-moving-average",
action="store_true",
help="Do not show moving average curve",
)
args = parser.parse_args()
with open(args.file) as f:
results = json.load(f)["results"]
label = results[0]["command"]
times = results[0]["times"]
num = len(times)
nums = range(num)
for result in results:
label = result["command"]
times = result["times"]
num = len(times)
nums = range(num)
plt.scatter(x=nums, y=times, marker=".", color="orange")
plt.ylim([0, None])
plt.xlim([-1, num])
plt.scatter(x=nums, y=times, marker=".")
plt.ylim([0, None])
plt.xlim([-1, num])
moving_average_width = (
num // 5 if args.moving_average_width is None else args.moving_average_width
)
if not args.no_moving_average:
moving_average_width = (
num // 5 if args.moving_average_width is None else args.moving_average_width
)
moving_average = moving_average(times, moving_average_width)
plt.plot(nums, moving_average, "-", color="blue")
average = moving_average(times, moving_average_width)
plt.plot(nums, average, "-")
if args.title:
plt.title(args.title)
plt.legend(labels=[label], loc="best", fontsize="medium")
legend = []
for result in results:
legend.append(result["command"])
if not args.no_moving_average:
legend.append("moving average")
plt.legend(legend)
plt.ylabel("Time [s]")
if args.output:
plt.savefig(args.output)
else: