diff --git a/graph_net_bench/aggregate_es_scores.py b/graph_net_bench/aggregate_es_scores.py index 45a2b5a4f..21f8a8a86 100644 --- a/graph_net_bench/aggregate_es_scores.py +++ b/graph_net_bench/aggregate_es_scores.py @@ -19,13 +19,14 @@ def main(args): for weight in [weights[tolerance]] for score in [es_scores[tolerance]] ) + rectified_speedup = 10 ** float(weighted_sum) result = { "id": args.sample_id, - "score": float(weighted_sum), + "score": rectified_speedup, } with open(args.output_json_file_path, "w") as f: json.dump(result, f, indent=4) - print(f"{weighted_sum=}") + print(f"{rectified_speedup=}") print(f"Result is saved to {args.output_json_file_path}")