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

python test + more models vis

Browse Source
This commit is contained in:
Lennart J. Kurzweg (Nx2)
2024-08-30 23:05:39 +02:00
parent 065fff21e6
commit 30b1cde43f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
visualize.py
View File

@@ -30,7 +30,7 @@ df = pd.DataFrame(results)
df['technique_name'] = pd.Categorical(df['technique_name'], categories=[techniques[1].name, techniques[572].name, techniques[903].name],ordered=True) df['technique_name'] = pd.Categorical(df['technique_name'], categories=[techniques[1].name, techniques[572].name, techniques[903].name],ordered=True)
df['test_name'] = pd.Categorical(df['test_name'], categories=[tests[607].name, tests[693].name, tests[120].name, tests[283].name, tests[260].name, tests[856].name],ordered=True) df['test_name'] = pd.Categorical(df['test_name'], categories=[tests[607].name, tests[693].name, tests[363].name, tests[120].name, tests[283].name, tests[260].name, tests[856].name],ordered=True)
sorted_df = df.sort_values('model_size') sorted_df = df.sort_values('model_size')
# Perform the groupby and unstack operation # Perform the groupby and unstack operation
@@ -96,7 +96,7 @@ pass_rate = pd.pivot_table(
pass_rate = pass_rate.loc[ordered_techniques] pass_rate = pass_rate.loc[ordered_techniques]
# Plot the heatmap # Plot the heatmap
plt.figure(figsize=(8, 10)) plt.figure(figsize=(8, 12))
sns.heatmap( sns.heatmap(
pass_rate * 100, pass_rate * 100,
annot=True, annot=True,