Commit Graph

3 Commits

Author SHA1 Message Date
czzhangheng abdd3165b8 修复STDEN模型bug:参数量异常和维度错误
问题分析:
1. 参数量异常小(16,522) - 缺少node到edge转换层
2. 维度错误 - 编码器期望edge格式但收到node格式输入
3. 解码器维度计算错误

修复内容:
- 添加node_to_edge和edge_to_node转换层,参数量从16,522增加到1,009,002
- 修改forward方法正确处理node格式输入输出
- 修复编码器以处理edge格式的中间数据
- 修正解码器中的维度计算问题

测试结果:
- 参数量:1,009,002 (合理范围)
- 输入输出形状正确:(batch_size, seq_len/horizon, num_nodes, input/output_dim)
- 模型可以正常前向传播
2025-09-11 12:39:46 +08:00
czzhangheng 626bb4d2bb 为STDEN模型添加node到edge的Linear转换层
- 在STDENModel中添加node_to_edge和edge_to_node转换层
- 修改forward方法以处理node_num输入并输出node_num格式
- 更新编码器以处理edge格式的中间数据
- 修复解码器中的维度计算问题
- 解决设备不匹配和数据类型不一致问题
- 更新.gitignore以允许models/STDEN/代码目录被跟踪

现在模型可以接受node_num格式的输入,内部转换为edge_num进行处理,最后转换回node_num输出。
2025-09-11 11:42:02 +08:00
czzhangheng ab5811425d add STDEN 2025-09-03 08:11:01 +08:00