Upload MTGNN model trained on METR-LA

README.md

@@ -21,10 +21,10 @@ This model uses a graph neural network architecture that combines:
 
 ## Evaluation Metrics
 
-- **Validation MAE**: 10.8337
-- **Test MAE (15 min)**: 11.0514
-- **Test MAPE (15 min)**: 0.3071
-- **Test RMSE (15 min)**: 13.7911
+- **Validation MAE**: 2.7572
+- **Test MAE (15 min)**: 6.6690
+- **Test MAPE (15 min)**: 0.2444
+- **Test RMSE (15 min)**: 12.8614
 
 
 ## Dataset

config.json

@@ -17,7 +17,7 @@
   "tanhalpha": 3.0,
   "batch_size": 64,
   "val_batch_size": 8,
-  "epochs": 1,
+  "epochs": 100,
   "learning_rate": 0.001,
   "weight_decay": 0.0001,
   "clip": 5,
@@ -26,9 +26,9 @@
   "gcn_true": true,
   "buildA_true": true,
   "cl": true,
-  "device": "mps",
+  "device": "cuda",
   "seed": 101,
   "print_every": 999999,
-  "save": "/Users/witold/Repos/witgaw/traffic-forecasting/models/mtgnn_metr-la_test/",
+  "save": "/workspace/traffic-forecasting/models/MTGNN/model_metr-la/",
   "dataset_name": "METR-LA"
 }

metadata.json

@@ -1,11 +1,11 @@
 {
   "dataset": "METR-LA",
-  "upload_date": "2025-11-10T11:30:56.415736",
+  "upload_date": "2025-11-10T16:03:05.929474",
   "metrics": {
-    "Validation MAE": 10.83370304107666,
-    "Test MAE (15 min)": 11.05143928527832,
-    "Test MAPE (15 min)": 0.307081401348114,
-    "Test RMSE (15 min)": 13.791097915462249
+    "Validation MAE": 2.757157325744629,
+    "Test MAE (15 min)": 6.668963432312012,
+    "Test MAPE (15 min)": 0.2443690299987793,
+    "Test RMSE (15 min)": 12.861386110584927
   },
   "framework": "PyTorch",
   "model_type": "MTGNN"

model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f0b80191818155f81cf57a152af28b8a3ed6ef6704d78a7435283660977f7bec
+oid sha256:228f517f73dd8099ff0cefd732420fe3fafc8aa42ed6f43397111bd7feee9c0b
 size 1801732

model.pth_config.json

@@ -17,7 +17,7 @@
   "tanhalpha": 3.0,
   "batch_size": 64,
   "val_batch_size": 8,
-  "epochs": 1,
+  "epochs": 100,
   "learning_rate": 0.001,
   "weight_decay": 0.0001,
   "clip": 5,
@@ -26,8 +26,8 @@
   "gcn_true": true,
   "buildA_true": true,
   "cl": true,
-  "device": "mps",
+  "device": "cuda",
   "seed": 101,
   "print_every": 999999,
-  "save": "/Users/witold/Repos/witgaw/traffic-forecasting/models/mtgnn_metr-la_test/"
+  "save": "/workspace/traffic-forecasting/models/MTGNN/model_metr-la/"
 }