update mlp for multitun

ChenS676 · Jul 2, 2024 · 66c0738 · 66c0738
1 parent a0c3a6c
commit 66c0738
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 65 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -30,8 +30,8 @@
 			"justMyCode": true,
 			"python.tensorboard.logDirectory": "/hkfs/work/workspace/scratch/cc7738-benchmark_tag/TAPE_chen/core/gcns/runs", 
 			"args": [
-				// "--data",
-				// "cora",
+				"--data",
+				"arxiv_2023",
 				// "--device",
 				// "cpu",
 				// "--epochs",

diff --git a/core/model_finetuning/mlp.py b/core/model_finetuning/mlp.py
@@ -169,14 +169,14 @@ def project_main():
 
         for key, result in results_rank.items():
             loggers[key].add_result(run_id, result)
-            
-        print(loggers[run_id].results)
+        st()
+
 
         for key in results_rank.keys():
-            print(loggers[key].calc_all_stats())
+            print(loggers[key].calc_run_stats(run_id))
         # clf = MLPClassifier(random_state=1, max_iter=300).fit(train_dataset, train_labels)
         # test_proba = clf.predict_proba(test_dataset)
         # test_pred = clf.predict(test_dataset)
         # acc = clf.score(test_dataset, test_labels)
 if __name__ == '__main__':
-    project_main()
+    project_main()
diff --git a/core/results/gae-gae-cora-origin/gae.yaml b/core/results/gae-gae-cora-origin/gae.yaml