Add model TESTAM

[hczs]

任务执行配置信息

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

{
  "batch_size": 16,
  "cache_dataset": true,
  "num_workers": 0,
  "pad_with_last_sample": false,

  "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
}

模型配置文件（TESTAM.json）

{
  "max_epoch": 100,

  "learner": "adam",
  "learning_rate": 1e-3,
  "lr_beta1": 0.9,
  "lr_beta2": 0.98,
  "lr_epsilon": 1e-9,

  "lr_decay": true,
  "lr_scheduler": "cosinewarmupscheduler",
  "n_warmup_steps": 4000,
  "lr_mul": 1,

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

  "clip_grad_norm": true,
  "max_grad_norm": 5,

  "dropout": 0.0,
  "prob_mul": false,
  "input_dim": 2,
  "hidden_size": 32,
  "layers": 3,
  "is_quantile": true,
  "quantile": 0.7
}

任务执行结果

模型评价指标对比如下

• METR_LA	METR_LA	LibCity	TESTAM
  MAE	3.46	3.36
  RMSE	7.31	7.09
  MAPE	10.06%	9.67%

• PEMS_BAY	PEMS_BAY	LibCity	TESTAM
  MAE	1.89	1.85
  RMSE	4.41	4.33
  MAPE	4.40%	4.31%