1 files changed, 2 insertions, 1 deletions
diff --git a/src/monkey_learns/linear_regression.zig b/src/monkey_learns/linear_regression.zig
index 4cc5a9f..17b886d 100644
--- a/src/monkey_learns/linear_regression.zig
+++ b/src/monkey_learns/linear_regression.zig
@@ -61,7 +61,7 @@ test "Linear Regression" {
     const y = [_]f64{ 3.1, 4.9, 7.2, 9.1, 11.0, 12.8, 14.9, 17.2, 18.8, 21.1 };
 
     const learning_rate: f64 = 0.01;
-    const epochs: usize = 1000;
+    const epochs: usize = 10000;
     model.train(&x, &y, learning_rate, epochs);
 
     try testing.expect(@abs(model.weight - 2.0) < 0.1);
@@ -72,6 +72,7 @@ test "Linear Regression" {
     for (test_x, expected_y) |xi, yi| {
         const prediction = model.predict(xi);
         try testing.expect(@abs(prediction - yi) < 0.5);
+        std.debug.print("Expected: {} | Predicted {}\n", .{ yi, prediction });
     }
 
     const mse = model.loss_function(&x, &y);