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 |
|
kingecg
|
16c2277474
|
```
feat(examples): 更新simple_model_example.go以使用gomatrix包
- 导入gomatrix包替代部分gotensor功能
- 修改权重张量梯度初始化使用gomatrix.NewZeros
- 更新must函数为泛型实现
- 重构损失函数实现使用gomatrix操作
- 优化输出格式化避免重复数据访问
refactor(model_test): 更新测试用例使用gomatrix构造函数
- 修改TestSequential测试使用gomatrix.NewMatrix和gomatrix.NewVector
- 更新TestSaveLoadModel测试使用gomatrix构造函数
- 修改TestLinearLayer测试使用NewTensor构造权重矩阵
refactor(trainer_test): 将Must函数改为泛型实现
- 更新Must函数为泛型版本支持任意类型
```
|
2026-01-01 15:14:10 +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 |