Changed to ridge model to reduce effects of overfitting on small models.

This commit is contained in:
Hunter Nichols 2021-02-26 11:00:21 -08:00
parent b5516865f1
commit d3ef1d7b85
1 changed files with 3 additions and 1 deletions

View File

@ -7,6 +7,7 @@
# #
from .regression_model import regression_model from .regression_model import regression_model
from sklearn.linear_model import Ridge
from globals import OPTS from globals import OPTS
import debug import debug
@ -23,7 +24,8 @@ class linear_regression(regression_model):
Supervised training of model. Supervised training of model.
""" """
model = LinearRegression() #model = LinearRegression()
model = Ridge()
model.fit(features, labels) model.fit(features, labels)
return model return model