From c125b9d46ee77055647bb69d8e19ab1c78fa739a Mon Sep 17 00:00:00 2001
From: smishra-SM <58541443+smishra-SM@users.noreply.github.com>
Date: Fri, 2 Oct 2020 00:11:57 +0530
Subject: [PATCH] Update regression.py

---
 regression.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/regression.py b/regression.py
index 88902cf..0841cc1 100755
--- a/regression.py
+++ b/regression.py
@@ -87,7 +87,8 @@ def cost(self, w, X, Y, lamb):
     Y_test = np.matrix(Y_test).T
     
     reg = LinearRegression()
-    reg.train(X_train, Y_train) 
+    reg.train(X_train, Y_train)  #you can use regularization parameter to reduce overfitting.
+
     reg.test(X_train, Y_train) 
     reg.test(X_test, Y_test)