Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Merged
merged 6 commits into from
Jul 2, 2024

Conversation

hczs
Copy link
Contributor

@hczs hczs commented Jun 21, 2024

本次修改三个问题:

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

@hczs
Copy link
Contributor Author

hczs commented Jun 24, 2024

最新提交

  1. 766c164
  2. 7f9f537

修改了两个问题

  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

@hczs hczs changed the title fix: TESTAM add vocab_size parameter and TimeMixer input adjust to the BTNF fix: use the time interval to calculate the TESTAM vocab szie parameter and TimeMixer input adjust to the BTNF Jun 25, 2024
@Kazeya27 Kazeya27 merged commit c92386c into LibCity:master Jul 2, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants