LibCity / Bigscity-LibCity

LibCity: An Open Library for Urban Spatial-temporal Data Mining
https://libcity.ai/
Apache License 2.0
937 stars 168 forks source link

Add model STPGCN #408

Closed hczs closed 7 months ago

hczs commented 7 months ago

任务执行配置信息

数据集配置文件(STPGCNDataset.json)

{
  "batch_size": 32,
  "cache_dataset": true,
  "train_rate": 0.6,
  "eval_rate": 0.2,
  "scaler": "standard",
  "load_external": false,
  "normal_external": false,
  "ext_scaler": "none",
  "pad_with_last_sample": false,
  "input_window": 12,
  "output_window": 12,

  "points_per_hour": 12,
  "alpha": 4,
  "beta": 2
}

模型配置文件(STPGCN.json)

{
  "max_epoch": 200,
  "learner": "adam",
  "learning_rate": 0.002,

  "use_early_stop": true,
  "patience": 150,

  "C": 64,
  "L": 3,
  "d": 8,
  "input_dim": 1,
  "output_dim": 1
}

任务执行结果

模型评价指标对比如下