Add version info to summary

Add repara version info tool
This commit is contained in:
2021-12-17 13:45:12 +01:00
parent cc14d338f8
commit fd3843cc8c
2 changed files with 24 additions and 3 deletions

View File

@@ -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(

20
src/repara.py Normal file
View File

@@ -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)