From 23e9cc015ebef8641f3b3fd0822d678374d247ce Mon Sep 17 00:00:00 2001 From: liwenyun Date: Sun, 22 Dec 2024 17:25:34 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20mse.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mse.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/mse.py b/mse.py index 7c7ab3f..8e73960 100644 --- a/mse.py +++ b/mse.py @@ -1,5 +1,8 @@ import matplotlib.pyplot as plt import numpy as np +import tikzplotlib + +# plt.style.use("ggplot") MCS = [ 'MI-FGSM', 'PGD', 'Adv-\nAttribute', 'Adv-\nDiffusion','TCA$^2$'] gold_medal = [ 54.82, 49.72, 50.12, 59.28,63.03] @@ -7,19 +10,21 @@ silver_medal = [ 31.06, 24.57, 18.63, 32.61,36.08] x = np.arange(len(MCS)) print(x) -width = 0.35 +width = 0.4 gold_x = x silver_x = x + width plt.bar(gold_x,gold_medal,width=width,color="royalblue", label='Face++') plt.bar(silver_x,silver_medal,width=width,color="tomato", label='Aliyun') # ​plt.bar(silver_x,silver_medal,width=width,color="silver") -plt.xticks(x+width, labels=MCS) +plt.xticks(x+width/2, labels=MCS) for i in range(len(MCS)): - plt.text(gold_x[i],gold_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=8) - plt.text(silver_x[i],silver_medal[i], silver_medal[i],va="bottom",ha="center",fontsize=8) + plt.text(gold_x[i],gold_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=10) + plt.text(silver_x[i],silver_medal[i], silver_medal[i],va="bottom",ha="center",fontsize=10) # plt.text(bronze_x[i],bronze_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=8) -plt.legend(loc=2,frameon=False) -plt.ylabel('Mean Confidence Scores') +plt.legend(loc=2,frameon=False, fontsize=10) +plt.ylabel('Mean Confidence Scores', fontsize=10) + +# tikzplotlib.save("test.tex") plt.show() # import matplotlib.pyplot as plt