def store_coeff(self,name,value,fnames=None,jsonformat=True): if jsonformat: d = {"coeffs":[round(x,3) for x in list(value)]} if fnames is not None: d["fnames"] = list(fnames) open(os.path.join(misc.coef_dir(),name+'.json'),'w').write(simplejson.dumps(d,indent=4)) else: colname1 = (name,'coeffs') colname2 = (name,'fnames') if colname1 in self.coeffs: del self.coeffs[colname1] if colname2 in self.coeffs: del self.coeffs[colname2] d = {colname1:value} if fnames is not None: d[colname2] = fnames self.coeffs = pd.concat([self.coeffs,pd.DataFrame(d)],axis=1)
def load_fnames(self,name,jsonformat=True): if jsonformat: d = simplejson.loads(open(os.path.join(misc.coef_dir(),name+'.json')).read()) return d["fnames"] else: return self.coeffs[(name,'fnames')].dropna()
def load_coeff(self,name,jsonformat=True): if jsonformat: d = simplejson.loads(open(os.path.join(misc.coef_dir(),name+'.json')).read()) return np.array(d["coeffs"]) else: return self.coeffs[(name,'coeffs')].dropna()