🌈 style(Redesign some output filename rules):

KarhouTam · Aug 10, 2024 · 79c51ac · 79c51ac
1 parent b517e19
commit 79c51ac
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/server/fedavg.py b/src/server/fedavg.py
@@ -780,8 +780,7 @@ def run(self):
             plt.ylabel("Accuracy")
             plt.legend()
             plt.savefig(
-            self.output_dir
-                / f"{self.args.common.dataset}.png",
+                self.output_dir / f"metrics.png",
                 bbox_inches="tight",
             )
 
@@ -810,8 +809,7 @@ def run(self):
                                 value=np.array(stats).T,
                             )
             df.to_csv(
-                self.output_dir
-                / f"{self.args.common.dataset}_acc_metrics.csv",
+                self.output_dir / f"metrics.csv",
                 index=True,
                 index_label="epoch",
             )