nx2/test-small-llms
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

prevent visualize from crashing

Browse Source 
If somehting exists in saved_results.json, that doesnt in the
suite_settings (comented out e.g.) it crashed. Now it doesnt.
This commit is contained in:
Lennart J. Kurzweg (Nx2)
2024-10-05 21:08:31 +02:00
parent 7e46382f32
commit c4c79c1d2a
1 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
visualize.py
View File

@@ -18,16 +18,19 @@ def get_df() -> pd.DataFrame:
data = json.load(f) data = json.load(f)
raw_data = [] raw_data = []
for test_hash, test_data in data.items(): for test_hash, test_data in data.items():
raw_data.append({ try:
"hash": test_hash, raw_data.append({
"model_name": models[test_data['model_id']].display_name, "hash": test_hash,
"model_size": models[test_data['model_id']].parameter_count_in_b, "model_name": models[test_data['model_id']].display_name,
"technique_name": techniques[test_data['technique_id']].name, "model_size": models[test_data['model_id']].parameter_count_in_b,
"model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}", "technique_name": techniques[test_data['technique_id']].name,
"seed": test_data['seed'], "model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}",
"test_name": tests[test_data['test_id']].name, "seed": test_data['seed'],
"validation": test_data['validation'] "test_name": tests[test_data['test_id']].name,
}) "validation": test_data['validation']
})
except KeyError:
pass
df = pd.DataFrame(raw_data) df = pd.DataFrame(raw_data)