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
15d0fb4
·
verified ·
1 Parent(s): 1716fe7

Upload STGFORMER_PRETRAIN model trained on METR-LA

Browse files
Files changed (6) hide show
  1. metadata.json +1 -1
  2. model.safetensors +1 -1
  3. pretrained_checkpoint/imputation_head.safetensors +1 -1
  4. pretrained_checkpoint/metadata.json +1 -1
  5. pretrained_checkpoint/model.safetensors +1 -1
  6. pretrained_checkpoint/pretrain_config.json +0 -1
metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": "METR-LA",
3
- "upload_date": "2025-12-08T00:58:01.654463",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAIN"
 
1
  {
2
  "dataset": "METR-LA",
3
+ "upload_date": "2025-12-08T19:46:49.017420",
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:8153aad6be768f269655d27cfe2c94f7fb2961f65c27c075353eee42508f8377
3
  size 7231112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7a2f7772ee512c2038bb795119c423f596e1a5462dd72a96c8f72e962bcb1e8
3
  size 7231112
pretrained_checkpoint/imputation_head.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7940f41ece420524f70244907be0ebddf8eec068740f559823077f4bd38b7818
3
  size 6328
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:50036a379862329809bd3e8464db0d29036f84344cea8842d5babc82be699b2b
3
  size 6328
pretrained_checkpoint/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "dataset": "METR-LA",
3
- "upload_date": "2025-12-08T00:42:28.489304",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAINED"
 
1
  {
2
  "dataset": "METR-LA",
3
+ "upload_date": "2025-12-08T19:31:14.508598",
4
  "metrics": {},
5
  "framework": "PyTorch",
6
  "model_type": "STGFORMER_PRETRAINED"
pretrained_checkpoint/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eb75d70cfebb79169a673cb166fa7d973704e41ebe2dec90ae5a571d46dfdff7
3
  size 7231112
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1dce52eed7327647591d337d673d02e07e09c11f820b3c80c3085e0458bdaa99
3
  size 7231112
pretrained_checkpoint/pretrain_config.json CHANGED
@@ -5,7 +5,6 @@
5
  "stage1_mask_ratio": 0.15,
6
  "stage2_epochs": 5,
7
  "stage2_mask_ratio": 0.1,
8
- "pretrain_window": 12,
9
  "learning_rate": 0.001,
10
  "pretrain_batch_size": 100,
11
  "imputation_iterations": 3,
 
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,