diff --git a/src/Results.py b/src/Results.py index 02dd1dc..643c38d 100644 --- a/src/Results.py +++ b/src/Results.py @@ -703,9 +703,10 @@ class Summary: print("*" + "-" * (length - 2) + "*") print("*" + whites(length - 2)) print( - f"* Model: {result.data['model']:15s} Score: " - f"{result.data['score_name']:17s} " - f"Metric: {metric:10.7f}" + whites(length - 69) + f"* Model: {result.data['model']:15s} " + f"Ver. {result.data['version']:10s} " + f"Score: {result.data['score_name']:10s} " + f"Metric: {metric:10.7f}" + whites(length - 78) ) print("*" + whites(length - 2)) print( diff --git a/src/repara.py b/src/repara.py new file mode 100644 index 0000000..ead437e --- /dev/null +++ b/src/repara.py @@ -0,0 +1,20 @@ +import os +import json + +from Experiments import Files, Folders + + +versions = dict(SVC="-", STree="1.2.3", ODTE="0.3.1") + +results = Files().get_all_results() +for result in results: + print(result) + file_name = os.path.join(Folders.results, result) + with open(file_name) as f: + data = json.load(f) + if "version" not in data: + print(f"Repairing version in {result}") + model = data["model"] + data["version"] = versions[model] if model in versions else "-" + with open(file_name, "w") as f: + json.dump(data, f, indent=4)