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

work in progress

Browse Source
This commit is contained in:
Till Raab 2023-06-16 15:37:54 +02:00
parent d11c83b46f
commit 988d80f7bd
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
ethogram.py
View File

@ -9,7 +9,7 @@ import networkx as nx
from IPython import embed
from event_time_correlations import load_and_converete_boris_events
glob_colors = ['#BA2D22', '#53379B', '#F47F17', '#3673A4', '#AAB71B', '#DC143C', '#1E90FF']
glob_colors = ['#BA2D22', '#53379B', '#F47F17', '#3673A4', '#AAB71B', '#DC143C', '#1E90FF', 'k']
def plot_transition_matrix(matrix, labels):
@ -139,9 +139,10 @@ def main(base_path):
for enu_tar, label_tar in enumerate(loop_labels):
n = len(event_times[:-1][(event_labels[1:] == label_tar) & (np.diff(event_times) > 5)])
marcov_matrix[-1, enu_tar] = n
marcov_matrix[-1, 5] = 0
loop_labels.append('void')
event_counts = np.append(event_counts, marcov_matrix[-1].sum())
embed()
quit()
### get those cases where ag_on does not point to event and no event points to corresponding ag_off ... add thise cases in marcov matrix
chase_on_idx = np.where(event_labels == loop_labels[4])[0]
chase_off_idx = np.where(event_labels == loop_labels[5])[0]

9
event_time_correlations.py
View File

@ -255,12 +255,14 @@ def main(base_path):
lose_chrips_centered_on_ag_on_t = []
lose_chrips_centered_on_contact_t = []
lose_chrips_centered_on_win_rises = []
lose_chrips_centered_on_win_chirp = []
lose_chirp_count = []
win_chrips_centered_on_ag_off_t = []
win_chrips_centered_on_ag_on_t = []
win_chrips_centered_on_contact_t = []
win_chrips_centered_on_lose_rises = []
win_chrips_centered_on_lose_chirp = []
win_chirp_count = []
lose_rises_centered_on_ag_off_t = []
@ -317,24 +319,29 @@ def main(base_path):
lose_chrips_centered_on_ag_on_t.append(event_centered_times(ag_on_off_t_GRID[:, 0], chirp_times[1]))
lose_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[1]))
lose_chrips_centered_on_win_rises.append(event_centered_times(rise_times[0], chirp_times[1]))
lose_chrips_centered_on_win_chirp.append(event_centered_times(chirp_times[0], chirp_times[1]))
lose_chirp_count.append(len(chirp_times[1]))
win_chrips_centered_on_ag_off_t.append(event_centered_times(ag_on_off_t_GRID[:, 1], chirp_times[0]))
win_chrips_centered_on_ag_on_t.append(event_centered_times(ag_on_off_t_GRID[:, 0], chirp_times[0]))
win_chrips_centered_on_contact_t.append(event_centered_times(contact_t_GRID, chirp_times[0]))
win_chrips_centered_on_lose_rises.append(event_centered_times(rise_times[1], chirp_times[0]))
win_chrips_centered_on_lose_chirp.append(event_centered_times(chirp_times[1], chirp_times[0]))
win_chirp_count.append(len(chirp_times[0]))
else:
lose_chrips_centered_on_ag_off_t.append(np.array([]))
lose_chrips_centered_on_ag_on_t.append(np.array([]))
lose_chrips_centered_on_contact_t.append(np.array([]))
lose_chrips_centered_on_win_rises.append(np.array([]))
lose_chrips_centered_on_win_chirp.append(np.array([]))
lose_chirp_count.append(np.nan)
win_chrips_centered_on_ag_off_t.append(np.array([]))
win_chrips_centered_on_ag_on_t.append(np.array([]))
win_chrips_centered_on_contact_t.append(np.array([]))
win_chrips_centered_on_lose_rises.append(np.array([]))
win_chrips_centered_on_lose_chirp.append(np.array([]))
win_chirp_count.append(np.nan)
# rises
@ -369,11 +376,13 @@ def main(base_path):
[lose_chrips_centered_on_ag_on_t, lose_chirp_count, r'chirp$_{lose}$ on chase$_{on}$'],
[lose_chrips_centered_on_contact_t, lose_chirp_count, r'chirp$_{lose}$ on contact'],
[lose_chrips_centered_on_win_rises, lose_chirp_count, r'chirp$_{lose}$ on rise$_{win}$'],
[lose_chrips_centered_on_win_chirp, lose_chirp_count, r'chirp$_{lose}$ on chirp$_{win}$'],
[win_chrips_centered_on_ag_off_t, win_chirp_count, r'chirp$_{win}$ on chase$_{off}$'],
[win_chrips_centered_on_ag_on_t, win_chirp_count, r'chirp$_{win}$ on chase$_{on}$'],
[win_chrips_centered_on_contact_t, win_chirp_count, r'chirp$_{win}$ on contact'],
[win_chrips_centered_on_lose_rises, win_chirp_count, r'chirp$_{win}$ on rise$_{lose}$'],
[win_chrips_centered_on_lose_chirp, win_chirp_count, r'chirp$_{win}$ on chirp$_{lose}$'],
[lose_rises_centered_on_ag_off_t, lose_rises_count, r'rise$_{lose}$ on chase$_{off}$'],
[lose_rises_centered_on_ag_on_t, lose_rises_count, r'rise$_{lose}$ on chase$_{on}$'],