From d3ef1d7b85226ae854c4cceefe3d2131ab00e1ab Mon Sep 17 00:00:00 2001
From: Hunter Nichols <hznichol@ucsc.edu>
Date: Fri, 26 Feb 2021 11:00:21 -0800
Subject: [PATCH] Changed to ridge model to reduce effects of overfitting on
 small models.

---
 compiler/characterizer/linear_regression.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/compiler/characterizer/linear_regression.py b/compiler/characterizer/linear_regression.py
index fac7a170..eec1c1a4 100644
--- a/compiler/characterizer/linear_regression.py
+++ b/compiler/characterizer/linear_regression.py
@@ -7,6 +7,7 @@
 #
 
 from .regression_model import regression_model
+from sklearn.linear_model import Ridge
 from globals import OPTS
 import debug
 
@@ -23,7 +24,8 @@ class linear_regression(regression_model):
         Supervised training of model.
         """
         
-        model = LinearRegression()
+        #model = LinearRegression()
+        model = Ridge()
         model.fit(features, labels)
         return model