search

LibCity / Bigscity-LibCity

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

fix: use the time interval to calculate the TESTAM vocab szie parameter and TimeMixer input adjust to the BTNF #426

Closed hczs closed 4 months ago

hczs commented 5 months ago

本次修改三个问题：

  1. TESTAM 模型添加 vocab_size 参数
  2. TimeMixer 移除无用的内部配置类
  3. TimeMixer 输入由 BTN 改为 BTNF 格式的数组
hczs commented 5 months ago

最新提交

  1. https://github.com/LibCity/Bigscity-LibCity/pull/426/commits/766c164546b1bb718113ab1baae5db1ed7c3ee04
  2. https://github.com/LibCity/Bigscity-LibCity/pull/426/commits/7f9f537daf472fdb8b67ed311e481c893ffe56e2

修改了两个问题

  1. TESTAM vocab_size 参数改为使用 time_intervals 参数来计算
  2. TimeMixer 输入输出的时候都添加一个线性层来处理feature问题

下面是运行结果指标 ==========PEMSD8========== libcity: MAE: 15.80, MAPE: 10.18%, RMSE: 24.98 paper: MAE: 15.22, MAPE: 9.67%, RMSE: 24.26 ==========PEMSD4-epoch30========== libcity: MAE: 19.40, MAPE: 13.35%, RMSE: 31.14 paper: MAE: 19.21, MAPE: 12.53%, RMSE: 30.92