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)
raw_data = []
for test_hash, test_data in data.items():
raw_data.append({
"hash": test_hash,
"model_name": models[test_data['model_id']].display_name,
"model_size": models[test_data['model_id']].parameter_count_in_b,
"technique_name": techniques[test_data['technique_id']].name,
"model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}",
"seed": test_data['seed'],
"test_name": tests[test_data['test_id']].name,
"validation": test_data['validation']
})
try:
raw_data.append({
"hash": test_hash,
"model_name": models[test_data['model_id']].display_name,
"model_size": models[test_data['model_id']].parameter_count_in_b,
"technique_name": techniques[test_data['technique_id']].name,
"model_technique": f"{models[test_data['model_id']].display_name}:{ techniques[test_data['technique_id']].name}",
"seed": test_data['seed'],
"test_name": tests[test_data['test_id']].name,
"validation": test_data['validation']
})
except KeyError:
pass
df = pd.DataFrame(raw_data)