Commit Graph

2 Commits

Author SHA1 Message Date
kingecg 28cf1f533a ```
test(model): 修复矩阵形状不匹配的测试用例

- 将bias从vector改为1x2矩阵以匹配输出形状
- 更新梯度初始化为矩阵格式而非向量
- 修改输入数据创建方式,使用NewMatrix替代NewVector
- 调整期望输出计算逻辑以正确处理矩阵运算

fix(optimizer): 修复Adam优化器在矩阵参数下的索引访问

- 添加对1维和2维形状的分别处理逻辑
- 修正一阶矩和二阶矩估计的索引访问方式
- 确保矩阵参数的每个元素都能正确更新

test(optimizer): 增强优化器测试覆盖矩阵参数场景

- 添加矩阵参数的Adam优化器测试用例
- 验证内部状态矩阵形状与参数保持一致
- 确保参数沿梯度反方向正确更新
```
2026-01-01 15:44:04 +08:00
程广 9aa53cbd5c ```
feat: 添加高级优化器示例和简单模型示例

添加了 advanced_optimizer_example.go 和 simple_model_example.go
两个示例文件,演示了不同优化器的使用方法和简单模型的训练过程。

同时添加了模型、优化器和训练器的单元测试文件,包括:
- model_test.go: 测试Sequential模型、模型保存加载功能和线性层
- optimizer_test.go: 测试SGD和Adam优化器功能
- trainer_test.go: 测试训练器的基本功能和完整训练过程

更新了go.mod和go.sum中的gomatrix依赖版本。
```
2025-12-31 17:52:34 +08:00