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 .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
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue