Upload MTGNN model trained on METR-LA

README.md

@@ -21,10 +21,11 @@ This model uses a graph neural network architecture that combines:
 
 ## Evaluation Metrics
 
-- **Validation MAE**: 2.7607
-- **Test MAE** (15 min): 2.4751
-- **Test MAPE** (15 min): 0.0623
-- **Test RMSE** (15 min): 4.6149
+- **Validation MAE**: 10.8337
+- **Test MAE (15 min)**: 11.0514
+- **Test MAPE (15 min)**: 0.3071
+- **Test RMSE (15 min)**: 13.7911
+
 
 ## Dataset
 

config.json

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

metadata.json

@@ -1,11 +1,11 @@
 {
   "dataset": "METR-LA",
-  "upload_date": "2025-11-07T12:28:24.692943",
+  "upload_date": "2025-11-10T11:30:56.415736",
   "metrics": {
-    "Validation MAE": 2.7607061862945557,
-    "Test MAE": 33.724578857421875,
-    "Test MAPE": Infinity,
-    "Test RMSE": 44.51845169067383
+    "Validation MAE": 10.83370304107666,
+    "Test MAE (15 min)": 11.05143928527832,
+    "Test MAPE (15 min)": 0.307081401348114,
+    "Test RMSE (15 min)": 13.791097915462249
   },
   "framework": "PyTorch",
   "model_type": "MTGNN"

model.pth

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

model.pth_config.json

@@ -0,0 +1,33 @@
+{
+  "num_nodes": 207,
+  "gcn_depth": 2,
+  "dropout": 0.3,
+  "subgraph_size": 20,
+  "node_dim": 40,
+  "dilation_exponential": 1,
+  "conv_channels": 32,
+  "residual_channels": 32,
+  "skip_channels": 64,
+  "end_channels": 128,
+  "in_dim": 2,
+  "seq_in_len": 12,
+  "seq_out_len": 12,
+  "layers": 3,
+  "propalpha": 0.05,
+  "tanhalpha": 3.0,
+  "batch_size": 64,
+  "val_batch_size": 8,
+  "epochs": 1,
+  "learning_rate": 0.001,
+  "weight_decay": 0.0001,
+  "clip": 5,
+  "step_size1": 2500,
+  "step_size2": 100,
+  "gcn_true": true,
+  "buildA_true": true,
+  "cl": true,
+  "device": "mps",
+  "seed": 101,
+  "print_every": 999999,
+  "save": "/Users/witold/Repos/witgaw/traffic-forecasting/models/mtgnn_metr-la_test/"
+}