Project-I/models
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
..
STDEN 修复STDEN模型bug:参数量异常和维度错误 2025-09-11 12:39:46 +08:00
STGODE 更新STGODE模型:添加配置文件、优化模型代码、新增测试数据文件 2025-09-03 10:18:52 +08:00
STGODE_LLM 添加STGODE_LLM和STGODE_LLM_GPT2模型实现,更新配置文件和README 2025-09-04 11:12:11 +08:00
STGODE_LLM_GPT2 优化READMER,注册STGODE-LLM 2025-09-04 06:05:09 +00:00
model_selector.py 优化READMER,注册STGODE-LLM 2025-09-04 06:05:09 +00:00