Project-I

Commit Graph

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
harry.zhang	70da9574da	优化READMER，注册STGODE-LLM	2025-09-04 06:05:09 +00:00
czzhangheng	e9e3da03d3	添加STGODE_LLM和STGODE_LLM_GPT2模型实现，更新配置文件和README	2025-09-04 11:12:11 +08:00
czzhangheng	387f64efab	更新STGODE模型：添加配置文件、优化模型代码、新增测试数据文件	2025-09-03 10:18:52 +08:00
czzhangheng	56a7f309fa	Merge STDEN-implementation branch and add STGODE models	2025-09-03 09:20:45 +08:00
czzhangheng	ab5811425d	add STDEN	2025-09-03 08:11:01 +08:00
czzhangheng	66a23ffbbb	更新项目文件：添加数据选择器、模型选择器、训练器选择器和ODE训练器	2025-09-03 07:52:18 +08:00
harry.zhang	19a02ba7ae	framework	2025-09-01 15:25:50 +08:00

9 Commits