czzhangheng
abdd3165b8
问题分析： 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) - 模型可以正常前向传播 |
||
|---|---|---|
| .. | ||
| STDEN_modules.md | ||
| diffeq_solver.py | ||
| ode_func.py | ||
| stden_model.py | ||
| utils.py | ||