jgrewe/gp_neurobio
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add chirpsize to dict

Browse Source
This commit is contained in:
Jan Grewe 2018-11-12 11:01:36 +01:00
parent 258b3074b8
commit e1add34c41
1 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
code/read_chirp_data.py
View File

@ -19,15 +19,17 @@ def load_chirp_spikes(dataset):
df = l.split(":")[-1]
if "contrast" in l and "true" not in l:
contrast = l.split(":")[-1]
if "chirpsize" in l:
cs = l.split(":")[-1]
if "#Key" in l:
spikes[(index, df, contrast)] = {}
spikes[(index, df, contrast, cs)] = {}
if "chirp index" in l:
ci = int(l.split(":")[-1])
if "beat phase" in l:
phase = float(l.split(":")[-1])
spikes[(index, df, contrast)][(ci, phase)] = []
spikes[(index, df, contrast, cs)][(ci, phase)] = []
if len(l.strip()) != 0 and "#" not in l:
spikes[(index, df, contrast)][(ci, phase)].append(float(l))
spikes[(index, df, contrast, cs)][(ci, phase)].append(float(l))
return spikes
@ -47,12 +49,15 @@ def load_chirp_times(dataset):
df = l.split(":")[-1]
if "contrast" in l and "true" not in l:
contrast = l.split(":")[-1]
if "chirpsize" in l:
cs = l.split(":")[-1]
if "#Key" in l:
chirp_times[(index, df, contrast)] = []
chirp_times[(index, df, contrast, cs)] = []
if len(l.strip()) != 0 and "#" not in l:
chirp_times[(index, df, contrast)].append(float(l.split()[1]))
chirp_times[(index, df, contrast, cs)].append(float(l.split()[1]))
return chirp_times
if __name__ == "__main__":
data_dir = "../data"
dataset = "2018-11-09-ad-invivo-1"