Method json2csv2

src/evaluate_prompt.py:1615–1642

(self, jsonfile)

    def json2csv2(self, jsonfile):
        results = json.load(open(jsonfile, "r"))
        lines = []
        for dataset in results:
            lines.append(dataset + "="*20)
            for metric in results[dataset]:
                lines.append(metric+ "+"*10)
                if metric in ["gt_speedup", "gt_opt"]:
                    lines.append("Strategy,Best@1,Best@5,Best@8,Best@10,Best@20,Avg")
                    for name in results[dataset][metric]:
                        line = "{},{},{},{},{},{},{}".format(name,
                        format(results[dataset][metric][name]["1"] * (100 if "speedup" not in metric else 1), ".2f"),
                        format(results[dataset][metric][name]["5"] * (100 if "speedup" not in metric else 1), ".2f"),
                        format(results[dataset][metric][name]["8"] * (100 if "speedup" not in metric else 1), ".2f"),
                        format(results[dataset][metric][name]["10"] * (100 if "speedup" not in metric else 1), ".2f"),
                        format(results[dataset][metric][name]["20"] * (100 if "speedup" not in metric else 1), ".2f"),
                        format(results[dataset][metric][name]["avg"] * (100 if "speedup" not in metric else 1), ".2f"))
                        lines.append(line)
                else:
                    lines.append("Strategy,Avg")
                    for name in results[dataset][metric]:
                        line = "{},{}".format(name,
                        format(results[dataset][metric][name] * (100 if "speedup" not in metric else 1), ".2f"))
                        lines.append(line)

        
        with open(jsonfile.replace(".json", ".csv"), "w", encoding = "utf-8") as f:
            f.write("\n".join(lines))