From 30b1cde43f609e5880d33da0bab1cc391736aca3 Mon Sep 17 00:00:00 2001 From: "Lennart J. Kurzweg (Nx2)" Date: Fri, 30 Aug 2024 23:05:39 +0200 Subject: [PATCH] python test + more models vis --- visualize.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/visualize.py b/visualize.py index ba31660..bc85598 100644 --- a/visualize.py +++ b/visualize.py @@ -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['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') # Perform the groupby and unstack operation @@ -96,7 +96,7 @@ pass_rate = pd.pivot_table( pass_rate = pass_rate.loc[ordered_techniques] # Plot the heatmap -plt.figure(figsize=(8, 10)) +plt.figure(figsize=(8, 12)) sns.heatmap( pass_rate * 100, annot=True,