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 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