feat: add Model MegaCRN

任务执行配置信息

数据集配置文件（TrafficStatePointDataset.json）

{
  "batch_size": 64,
  "cache_dataset": true,
  "num_workers": 0,
  "pad_with_last_sample": true,
  "train_rate": 0.7,
  "eval_rate": 0.1,
  "scaler": "standard",
  "load_external": true,
  "normal_external": false,
  "ext_scaler": "none",
  "input_window": 12,
  "output_window": 12,
  "add_time_in_day": true,
  "add_day_in_week": false
}

模型配置文件（MegaCRN.json）

{
  "rnn_units": 64,
  "num_layers": 1,
  "cheb_k": 3,
  "ycov_dim": 1,
  "cl_decay_steps": 2000,
  "use_curriculum_learning": true,
  "mem_num": 20,
  "mem_dim": 64,
  "lamb": 0.01,
  "lamb1": 0.01
}

执行器配置信息（MegaCRNExecutor.json）

{
  "max_epoch": 200,
  "learner": "adam",
  "learning_rate": 0.01,
  "lr_epsilon": 0.001,
  "lr_decay": true,
  "lr_scheduler": "multisteplr",
  "lr_decay_ratio": 0.1,
  "milestones": [50,100],
  "clip_grad_norm": true,
  "max_grad_norm": 5,
  "use_early_stop": true,
  "patience": 20
}

任务执行结果

模型评价指标对比如下

METR_LA	METR_LA	MAE/RMSE/MAPE@3rd step	MAE/RMSE/MAPE@6th step	MAE/RMSE/MAPE@12th step
MegaCRN	2.52/4.94/6.44%	2.93/6.06/7.96%	3.38/7.23/9.72%
LibCity	2.67/5.07/6.88%	3.08/6.18/8.36%	3.55/7.38/10.18%

PEMS_BAY	PEMS_BAY	MAE/RMSE/MAPE@3rd step	MAE/RMSE/MAPE@6th step	MAE/RMSE/MAPE@12th step
MegaCRN	1.28/2.72/2.67%	1.60/3.68/3.57%	1.88/4.42/4.41%
LibCity	1.34/2.85/2.79%	1.69/3.81/3.80%	2.01/4.56/4.78%

LibCity / Bigscity-LibCity

feat: add Model MegaCRN #427

任务执行配置信息

任务执行结果