mirror of
https://github.com/Magnus167/rustframe.git
synced 2025-08-19 23:20:01 +00:00
Increase training epochs from 5000 to 10000 for improved model performance
This commit is contained in:
parent
70d2a7a2b4
commit
75d07371b2
@ -263,7 +263,7 @@ mod tests {
|
||||
initializer: InitializerKind::He,
|
||||
loss: LossKind::BCE,
|
||||
learning_rate: 0.01,
|
||||
epochs: 5000,
|
||||
epochs: 10000,
|
||||
};
|
||||
let mut model = DenseNN::new(config);
|
||||
let x = Matrix::from_vec(vec![1.0, 2.0, 3.0, 4.0], 4, 1);
|
||||
|
Loading…
x
Reference in New Issue
Block a user