diff --git a/runner/eval_thresholds.py b/runner/eval_thresholds.py index db2893e..ffa8b7e 100644 --- a/runner/eval_thresholds.py +++ b/runner/eval_thresholds.py @@ -92,8 +92,7 @@ def pass_time(tensorboard_logfile: PurePath) -> float: ranks: dict[str, int] = defaultdict(int) for dataset_name, time_algos in timings.items(): time_algos.sort() - print(dataset_name) - print("=" * len(dataset_name)) + print("##", dataset_name) N = len(time_algos) rank = 1 for t, algo_name in time_algos: @@ -102,11 +101,10 @@ def pass_time(tensorboard_logfile: PurePath) -> float: else: _rank = rank rank += 1 - print(f"{_rank}: {algo_name}") + print(f"- {_rank}: {algo_name}") ranks[algo_name] += _rank print("\n") - print("Leaderboard") - print("===========") + print("## Leaderboard") for algo_name, _ in sorted(ranks.items(), key=lambda algo_rank: algo_rank[1]): - print(algo_name) + print("-", algo_name)