From 90b231490df2a04d46090f261ba2d3e31cff5e61 Mon Sep 17 00:00:00 2001
From: Athmane Mansour Bahar <ja_mansourbahar@esi.dz>
Date: Thu, 15 Aug 2024 17:51:01 +0000
Subject: [PATCH] Upload New File

---
 trainer/utils/config.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)
 create mode 100644 trainer/utils/config.py

diff --git a/trainer/utils/config.py b/trainer/utils/config.py
new file mode 100644
index 0000000..bb9c1ef
--- /dev/null
+++ b/trainer/utils/config.py
@@ -0,0 +1,20 @@
+import argparse
+
+
+def build_args():
+    parser = argparse.ArgumentParser(description="MAGIC")
+    parser.add_argument("--dataset", type=str, default="wget")
+    parser.add_argument("--device", type=int, default=-1)
+    parser.add_argument("--lr", type=float, default=0.001,
+                        help="learning rate")
+    parser.add_argument("--weight_decay", type=float, default=5e-4,
+                        help="weight decay")
+    parser.add_argument("--negative_slope", type=float, default=0.2,
+                        help="the negative slope of leaky relu for GAT")
+    parser.add_argument("--mask_rate", type=float, default=0.5)
+    parser.add_argument("--alpha_l", type=float, default=3, help="`pow`inddex for `sce` loss")
+    parser.add_argument("--optimizer", type=str, default="adam")
+    parser.add_argument("--loss_fn", type=str, default='sce')
+    parser.add_argument("--pooling", type=str, default="mean")
+    args = parser.parse_args()
+    return args
-- 
GitLab