mirror of https://github.com/VLSIDA/OpenRAM.git
Changed to ridge model to reduce effects of overfitting on small models.
This commit is contained in:
parent
b5516865f1
commit
d3ef1d7b85
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue