mirror of
https://github.com/Doctorado-ML/benchmark.git
synced 2025-08-16 07:55:54 +00:00
Add version info to summary
Add repara version info tool
This commit is contained in:
@@ -703,9 +703,10 @@ class Summary:
|
|||||||
print("*" + "-" * (length - 2) + "*")
|
print("*" + "-" * (length - 2) + "*")
|
||||||
print("*" + whites(length - 2))
|
print("*" + whites(length - 2))
|
||||||
print(
|
print(
|
||||||
f"* Model: {result.data['model']:15s} Score: "
|
f"* Model: {result.data['model']:15s} "
|
||||||
f"{result.data['score_name']:17s} "
|
f"Ver. {result.data['version']:10s} "
|
||||||
f"Metric: {metric:10.7f}" + whites(length - 69)
|
f"Score: {result.data['score_name']:10s} "
|
||||||
|
f"Metric: {metric:10.7f}" + whites(length - 78)
|
||||||
)
|
)
|
||||||
print("*" + whites(length - 2))
|
print("*" + whites(length - 2))
|
||||||
print(
|
print(
|
||||||
|
20
src/repara.py
Normal file
20
src/repara.py
Normal 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)
|
Reference in New Issue
Block a user