LibCity / Bigscity-LibCity

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

Add model RGSL #404

Closed hczs closed 6 months ago

hczs commented 6 months ago

任务执行配置信息

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

{
  "batch_size": 64,
  "cache_dataset": false,
  "num_workers": 0,
  "train_rate": 0.6,
  "eval_rate": 0.2,
  "scaler": "standard",
  "load_external": false,
  "normal_external": false,
  "ext_scaler": "none",
  "input_window": 12,
  "output_window": 12
}

模型文件配置(RGSL.json) 其中embed_dim参数在使用PEMSD4数据集进行训练时,需要将此参数改为10

{
  "max_epoch": 100,

  "learner": "adam",
  "learning_rate": 0.003,
  "lr_epsilon": 1.0e-8,
  "weight_decay": 0,

  "use_early_stop": true,
  "patience": 15,

  "input_dim": 1,
  "output_dim": 1,
  "rnn_units": 64,
  "num_layers": 2,
  "embed_dim": 8,
  "cheb_k": 2
}

任务执行结果

模型评价指标对比如下