Closed hczs closed 8 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
embed_dim
PEMSD4
{ "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 }
模型评价指标对比如下
PEMSD8
任务执行配置信息
数据集文件配置(RGSLDataset.json)
模型文件配置(RGSL.json) 其中
embed_dim
参数在使用PEMSD4
数据集进行训练时,需要将此参数改为10任务执行结果
模型评价指标对比如下
PEMSD4
PEMSD8