Hugging Face's logo

witgaw
/
STGFORMER_PRETRAIN_METR-LA

Safetensors
traffic-forecasting
time-series
graph-neural-network
stgformer_pretrain
Model card Files
xet
 Community
witgaw commited on
Commit
0b2775c
·
verified ·
1 Parent(s): 15d0fb4

Upload STGFORMER_PRETRAIN model trained on METR-LA

Browse files
Files changed (5) hide show
  1. config.json +1 -0
  2. metadata.json +1 -1
  3. model.safetensors +1 -1
  4. pretrained_checkpoint/metadata.json +1 -1
  5. pretrained_checkpoint/pretrain_config.json +4 -3
config.json CHANGED
@@ -14,6 +14,7 @@
14
  "num_layers": 3,
15
  "dropout_a": 0.3,
16
  "use_mixed_proj": true,
 
17
  "model_dim": 128,
18
  "dataset": "METR-LA",
19
  "graph_mode": "learned",
 
14
  "num_layers": 3,
15
  "dropout_a": 0.3,
16
  "use_mixed_proj": true,
17
+ "pre_attn_kernel_size": 1,
18
  "model_dim": 128,
19
  "dataset": "METR-LA",
20
  "graph_mode": "learned",
metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": "METR-LA",
3
- "upload_date": "2025-12-08T19:46:49.017420",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAIN"
 
1
  {
2
  "dataset": "METR-LA",
3
+ "upload_date": "2025-12-10T10:49:43.126314",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAIN"
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d7a2f7772ee512c2038bb795119c423f596e1a5462dd72a96c8f72e962bcb1e8
3
  size 7231112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46064684cc6ae5374c9927d582755e30a6f140ec82bef22f3dc0e4967fe2694f
3
  size 7231112
pretrained_checkpoint/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": "METR-LA",
3
- "upload_date": "2025-12-08T19:31:14.508598",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAINED"
 
1
  {
2
  "dataset": "METR-LA",
3
+ "upload_date": "2025-12-10T10:34:16.497730",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAINED"
pretrained_checkpoint/pretrain_config.json CHANGED
@@ -5,10 +5,11 @@
5
  "stage1_mask_ratio": 0.15,
6
  "stage2_epochs": 5,
7
  "stage2_mask_ratio": 0.1,
8
- "learning_rate": 0.001,
9
  "pretrain_batch_size": 100,
10
- "imputation_iterations": 3,
11
- "save_to": "STGFORMER_PRETRAINED"
 
12
  },
13
  "model_dim": 128,
14
  "num_nodes": 207
 
5
  "stage1_mask_ratio": 0.15,
6
  "stage2_epochs": 5,
7
  "stage2_mask_ratio": 0.1,
8
+ "use_normalized_data": false,
9
  "pretrain_batch_size": 100,
10
+ "learning_rate": 0.001,
11
+ "save_to": "STGFORMER_PRETRAINED",
12
+ "imputation_iterations": 3
13
  },
14
  "model_dim": 128,
15
  "num_nodes": 207