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