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
程广 fcc2e54144 ```
feat: 添加模型定义、优化器和训练器功能

- 实现Model接口和Sequential序列模型,支持模型定义和前向传播
- 添加SGD和Adam优化器,支持参数更新和梯度清零
- 创建Trainer训练器,提供完整的训练和评估流程
- 实现模型保存和加载功能,支持参数序列化
- 更新README文档,添加模型训练示例和API文档
- 重构README中的功能特性和示例代码
```
2025-12-31 14:45:29 +08:00