Upload MTGNN model trained on PEMS-BAY

README.md

@@ -21,10 +21,10 @@ This model uses a graph neural network architecture that combines:
 
 ## Evaluation Metrics
 
-- **Validation MAE**: 5.6602
-- **Test MAE (15 min)**: 14.3484
-- **Test MAPE (15 min)**: 0.2639
-- **Test RMSE (15 min)**: 20.4166
+- **Validation MAE**: 1.5766
+- **Test MAE (15 min)**: 54.1346
+- **Test MAPE (15 min)**: 0.8623
+- **Test RMSE (15 min)**: 54.9303
 
 
 ## 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_pems-bay_test/",
+  "save": "/workspace/traffic-forecasting/models/MTGNN/model_pems-bay/",
   "dataset_name": "PEMS-BAY"
 }

metadata.json

@@ -1,11 +1,11 @@
 {
   "dataset": "PEMS-BAY",
-  "upload_date": "2025-11-10T11:35:27.280302",
+  "upload_date": "2025-11-10T20:27:34.466045",
   "metrics": {
-    "Validation MAE": 5.660176753997803,
-    "Test MAE (15 min)": 14.348429679870605,
-    "Test MAPE (15 min)": 0.2638642191886902,
-    "Test RMSE (15 min)": 20.41659998461156
+    "Validation MAE": 1.5765657424926758,
+    "Test MAE (15 min)": 54.134620666503906,
+    "Test MAPE (15 min)": 0.8623476624488831,
+    "Test RMSE (15 min)": 54.9303336302846
   },
   "framework": "PyTorch",
   "model_type": "MTGNN"

model.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:96926806af5cec3a6308ee7cd4a66c783939b78c829b3608d9150938323a9dda
+oid sha256:34426775ecdc354eb746f0782954ec2048c6d2f6d8e9a0d004a8acaedcdf7867
 size 2724964

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_pems-bay_test/"
+  "save": "/workspace/traffic-forecasting/models/MTGNN/model_pems-bay/"
 }