From 75d07371b2edb3ce6b366cd8cf3df112ba82dd6a Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Sun, 6 Jul 2025 20:16:06 +0100
Subject: [PATCH] Increase training epochs from 5000 to 10000 for improved
 model performance

---
 src/compute/models/dense_nn.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/compute/models/dense_nn.rs b/src/compute/models/dense_nn.rs
index 7be04a2..413ec5c 100644
--- a/src/compute/models/dense_nn.rs
+++ b/src/compute/models/dense_nn.rs
@@ -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);