LibCity / Bigscity-LibCity

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

ESG #383

Closed Kazeya27 closed 7 months ago

Kazeya27 commented 7 months ago

原论文用的是2016年4月到6月的NYC-bike/taxi数据集,我在云盘里没找到。

Dataset Horizon-3 Horizon-6 Horizon-12
NYC-Bike RMSE=2.5529--MAE=1.5483 RMSE=2.6484--MAE=1.6026 RMSE=2.8778--MAE=1.7173
NYC-Taxi RMSE=8.5745--MAR=4.8750 RMSE=9.0125--MAE=5.0500 RMSE=9.7857--MAE=5.4019
Dataset Horizon-3 Horizon-6 Horizon-12
NYCBike20160809 RMSE=4.823869--MAE=2.041650 RMSE=4.975906--MAE=2.106852 RMSE=5.341728--MAE=2.222684
NYCTaxi20160102 RMSE=12.480914--MAE=4.200171 RMSE=13.343316--MAE=4.392735 RMSE=14.703619--MAE=4.710590
Kazeya27 commented 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 }