magnus167/rustframe
1
0
Fork 0
mirror of https://github.com/Magnus167/rustframe.git synced 2025-08-20 04:00:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increase training epochs from 5000 to 10000 for improved model performance

Browse Source
This commit is contained in:
Palash Tyagi 2025-07-06 20:16:06 +01:00
parent 70d2a7a2b4
commit 75d07371b2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/compute/models/dense_nn.rs
View File

@ -263,7 +263,7 @@ mod tests {
initializer: InitializerKind::He, initializer: InitializerKind::He,
loss: LossKind::BCE, loss: LossKind::BCE,
learning_rate: 0.01, learning_rate: 0.01,
epochs: 5000, epochs: 10000,
}; };
let mut model = DenseNN::new(config); let mut model = DenseNN::new(config);
let x = Matrix::from_vec(vec![1.0, 2.0, 3.0, 4.0], 4, 1); let x = Matrix::from_vec(vec![1.0, 2.0, 3.0, 4.0], 4, 1);