mirror of
https://github.com/VLSIDA/OpenRAM.git
synced 2026-09-02 11:08:01 +02:00
Changed to ridge model to reduce effects of overfitting on small models.
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user