Closed Kazeya27 closed 7 months ago
参数与论文中用于NYC数据集的一致 { "max_epoch": 300,
"learner": "adam", "learning_rate": 0.001, "lr_epsilon": 1e-8, "weight_decay": 0.0001,
"clip_grad_norm": true, "max_grad_norm": 5, "use_early_stop": true, "patience": 30,
"n_layers": 3, "dropout": 0.3, "residual_channels": 32, "end_channels": 128, "n_blocks": 1, "st_embedding_dim": 40, "kernel_set": [2, 6], "dilation_exp": 1, "conv_channels": 32, "gcn_depth": 2, "dy_embedding_dim": 20, "skip_channels": 64, "layer_norm_affline": false, "propalpha": 0.05, "dy_interval": [1, 1, 1] }
{ "batch_size": 16, "cache_dataset": false, "num_workers": 0, "pad_with_last_sample": true, "train_rate": 0.7, "eval_rate": 0.15, "scaler": "standard", "load_external": false, "normal_external": false, "ext_scaler": "none", "input_window": 12, "output_window": 12, "add_time_in_day": false, "add_day_in_week": false, "use_row_column": false }
原论文用的是2016年4月到6月的NYC-bike/taxi数据集,我在云盘里没找到。