raab/GP2023_chirp_detection
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

tryingen to get behaviour data

Browse Source
This commit is contained in:
wendtalexander 2023-01-19 15:13:30 +01:00
parent ef0c92bb61
commit 90c6a15433
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
code/behavior.py
View File

@ -32,6 +32,7 @@ class Behavior:
LED_on_time_BORIS = np.load(os.path.join(folder_path, 'LED_on_time.npy'), allow_pickle=True) LED_on_time_BORIS = np.load(os.path.join(folder_path, 'LED_on_time.npy'), allow_pickle=True)
self.time = np.load(datapath + "times.npy", allow_pickle=True) self.time = np.load(datapath + "times.npy", allow_pickle=True)
self.dataframe = read_csv(os.path.join(folder_path, self.file[1:-7] + '.csv')) self.dataframe = read_csv(os.path.join(folder_path, self.file[1:-7] + '.csv'))
for k, key in enumerate(self.dataframe.keys()): for k, key in enumerate(self.dataframe.keys()):
key = key.lower() key = key.lower()
if ' ' in key: if ' ' in key:
@ -83,5 +84,5 @@ def main(datapath: str):
if __name__ == '__main__': if __name__ == '__main__':
# Path to the data # Path to the data
datapath = '../data/mount_data/2020-03-13-10_00/' datapath = '../data/mount_data/2020-05-13-10_00/'
main(datapath) main(datapath)