 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
|
e65be7d668
|
更新.gitignore以屏蔽所有数据集文件类型
|
2025-09-11 11:27:05 +08:00 |
|
czzhangheng
|
cd542e18ef
|
合并远程更改并解决冲突
|
2025-09-11 11:16:34 +08:00 |
harry.zhang
|
6f0953b4ae
|
Remove .idea directory from tracking and enable .idea/ in .gitignore
|
2025-09-04 06:13:54 +00:00 |
|
harry.zhang
|
c1a4962795
|
README FIX
|
2025-09-04 06:11:50 +00:00 |
|
harry.zhang
|
167a63c65b
|
Remove PEMS08 from tracking and update .gitignore to ignore PEMS08/PEMSD8 datasets
|
2025-09-04 06:09:17 +00:00 |
|
harry.zhang
|
70da9574da
|
优化READMER，注册STGODE-LLM
|
2025-09-04 06:05:09 +00:00 |
|
harry.zhang
|
8ee9b4cfb2
|
Update README
|
2025-09-04 03:55:23 +00:00 |
|
harry.zhang
|
d62319302f
|
update_README
|
2025-09-04 03:53:02 +00:00 |
|
czzhangheng
|
0bf678f290
|
屏蔽pre-trained文件夹，从版本控制中移除预训练模型文件
|
2025-09-04 11:22:02 +08:00 |
|
czzhangheng
|
42e250d8e1
|
添加requirements.txt依赖文件，更新README和utils/download.py，移除STDEN子模块
|
2025-09-04 11:20:17 +08:00 |
|
czzhangheng
|
e9e3da03d3
|
添加STGODE_LLM和STGODE_LLM_GPT2模型实现，更新配置文件和README
|
2025-09-04 11:12:11 +08:00 |
|
czzhangheng
|
f0d3460c89
|
添加models/gpt2/到.gitignore以屏蔽GPT2模型文件
|
2025-09-04 11:06:07 +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
|
a8cc3a20fd
|
屏蔽exp和STDEN文件夹：从Git跟踪中移除但保留本地文件
|
2025-09-03 08:15:30 +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 |
|
czzhangheng
|
df8c573f4c
|
add STDEN
|
2025-09-03 07:51:30 +08:00 |
|
harry.zhang
|
19a02ba7ae
|
framework
|
2025-09-01 15:25:50 +08:00 |
|
czzhangheng
|
568fff7e99
|
Initial commit
|
2025-09-01 11:03:24 +08:00 |