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

sort chirps and chirp ids in time

Browse Source
This commit is contained in:
sprause 2023-01-19 16:14:00 +01:00
parent 8d790ad89c
commit 9bc13294d8
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
code/behavior.py
View File

@ -53,8 +53,6 @@ class Behavior:
self.start_s = (self.start_s - shift) / factor
self.stop_s = (self.stop_s - shift) / factor
"""
1 - chasing onset
2 - chasing offset
@ -83,9 +81,14 @@ temporal encpding needs to be corrected ... not exactly 25FPS.
"""
def main(datapath: str):
# behabvior is pandas dataframe with all the data
# behavior is pandas dataframe with all the data
bh = Behavior(datapath)
# chirps are not sorted in time (presumably due to prior groupings)
# sort chirps and corresponding fish_ids of the chirps
bh.chirps = bh.chirps[np.argsort(bh.chirps)]
bh.chirps_ids = bh.chirps_ids[np.argsort(bh.chirps)]
fig, ax = plt.subplots()
ax.plot(bh.chirps, np.ones_like(bh.chirps), marker='o')