def save_dic(self, d, **k): print 'Saving ' + self.file_name for keys, data in misc.dict_iter(d): if not misc.dict_haskey(self.dic, keys): self.add_storage(keys) s = misc.dict_recursive_get(self.dic, keys) s.save_data('-'.join(keys), data, **k) self.save()
def save_dic(self, d, **k): print 'Saving '+self.file_name for keys, data in misc.dict_iter(d): if not misc.dict_haskey(self.dic, keys): self.add_storage(keys) s=misc.dict_recursive_get(self.dic, keys) s.save_data('-'.join(keys), data, **k) self.save()
def postprocess(d): out = {} ylabels0 = sorted(d.keys()) xlabels = sorted(d[ylabels0[0]]['Net_0'].keys()) i, j = 0, 0 for tp in ['beta', 'sw']: ylabels = [] for key in ylabels0: if key[0] != tp[0] or len(d[key].keys()) < 2: continue ylabels.append(key) j += 1 # if j==396: # pass for net in d[key].keys(): mrs = [] for model in xlabels: obj = d[key][net][model]['spike_statistic'] mrs.append(obj.rates['mean']) keys = [tp, net, 'mean_rates'] if not misc.dict_haskey(out, keys): misc.dict_recursive_add(out, keys, [mrs]) if net == 'Net_0': i += 1 else: out[tp][net]['mean_rates'].append(mrs) if net == 'Net_0': i += 1 if not i == j: print d[key].keys(), i, j raise # print j,i for net in d[key].keys(): keys = [tp, net, 'xlabels'] misc.dict_recursive_add(out, keys, xlabels) keys = [tp, net, 'ylabels'] misc.dict_recursive_add(out, keys, ylabels) pp(out) for tp in out.keys(): for net in out[tp].keys(): out[tp][net]['mean_rates'] = numpy.array( out[tp][net]['mean_rates']) return out
def postprocess(d): out={} ylabels0=sorted(d.keys()) xlabels=sorted(d[ylabels0[0]]['Net_0'].keys()) i,j=0,0 for tp in ['beta','sw']: ylabels=[] for key in ylabels0: if key[0]!=tp[0] or len(d[key].keys())<2: continue ylabels.append(key) j+=1 # if j==396: # pass for net in d[key].keys(): mrs=[] for model in xlabels: obj=d[key][net][model]['spike_statistic'] mrs.append(obj.rates['mean']) keys=[tp, net, 'mean_rates'] if not misc.dict_haskey(out, keys): misc.dict_recursive_add(out, keys, [mrs]) if net=='Net_0': i+=1 else: out[tp][net]['mean_rates'].append(mrs) if net=='Net_0':i+=1 if not i==j: print d[key].keys(), i,j raise # print j,i for net in d[key].keys(): keys=[tp, net, 'xlabels'] misc.dict_recursive_add(out, keys, xlabels) keys=[tp, net, 'ylabels'] misc.dict_recursive_add(out, keys, ylabels) pp(out) for tp in out.keys(): for net in out[tp].keys(): out[tp][net]['mean_rates']=numpy.array(out[tp][net]['mean_rates']) return out
def process_exp(exp, tr): out = {} put_at = {'STN': 3, 'TA': 0, 'TI': 1, 'all': 2} for keys, val in misc.dict_iter(exp): if keys[-1] == 'CV': continue keys2 = [tr[keys[1]], tr[keys[2]], 'mean_rates'] if not misc.dict_haskey(out, keys2): a = numpy.zeros(4) a[put_at[keys[0]]] = val misc.dict_recursive_add(out, keys2, a) else: out[tr[keys[1]]][tr[keys[2]]]['mean_rates'][put_at[keys[0]]] = val pp(out) return out
def process_exp(exp, tr): out={} put_at={'STN':3, 'TA':0, 'TI':1, 'all':2} for keys, val in misc.dict_iter(exp): if keys[-1]=='CV': continue keys2=[tr[keys[1]],tr[keys[2]], 'mean_rates'] if not misc.dict_haskey(out, keys2): a=numpy.zeros(4) a[put_at[keys[0]]]=val misc.dict_recursive_add(out, keys2,a ) else: out[tr[keys[1]]] [tr[keys[2]]] ['mean_rates'][put_at[keys[0]]]=val pp(out) return out