luke
/
OpenRAM
mirror of https://github.com/VLSIDA/OpenRAM.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

linear regression model coefficients are now written to the extended config file

This commit is contained in:
Hunter Nichols 2021-03-02 13:14:56 -08:00
parent d3ef1d7b85
commit 2cd3d28add
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
compiler/characterizer/regression_model.py
View File

@ -137,5 +137,16 @@ class regression_model(simulation):
features, labels = get_scaled_data(dpath)
model = self.generate_model(features, labels)
models[dname] = model
self.save_model(dname, model)
return models
# Fixme - only will work for sklearn regression models
def save_model(self, model_name, model):
try:
OPTS.model_dict
except AttributeError:
OPTS.model_dict = {}
OPTS.model_dict[model_name+"_coef"] = list(model.coef_[0])
debug.info(1,"Coefs of {}:{}".format(model_name,OPTS.model_dict[model_name+"_coef"]))
OPTS.model_dict[model_name+"_intercept"] = float(model.intercept_)