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

plot

Browse Source
This commit is contained in:
Till Raab 2023-05-26 15:33:23 +02:00
parent 8a9ec1beed
commit aec5e298bc
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
event_time_analysis.py
View File

@ -91,6 +91,7 @@ def kde(event_dt, max_dt = 60):
conv_array += gauss(conv_t, e, kernal_w, kernal_h, norm=True) conv_array += gauss(conv_t, e, kernal_w, kernal_h, norm=True)
# plt.plot(conv_t, conv_array) # plt.plot(conv_t, conv_array)
return conv_array
def permulation_kde(event_dt, repetitions = 2000, max_dt = 60, max_mem_use_GB = 4, norm_count = 1): def permulation_kde(event_dt, repetitions = 2000, max_dt = 60, max_mem_use_GB = 4, norm_count = 1):
@ -119,8 +120,6 @@ def permulation_kde(event_dt, repetitions = 2000, max_dt = 60, max_mem_use_GB =
del event_dt_perm, gauss_3d del event_dt_perm, gauss_3d
return kde_3d return kde_3d
embed()
quit()
t0 = time.time() t0 = time.time()
kernal_w = 1 kernal_w = 1
kernal_h = 0.2 kernal_h = 0.2
@ -157,11 +156,16 @@ def permulation_kde(event_dt, repetitions = 2000, max_dt = 60, max_mem_use_GB =
# del event_dt_perm, gauss_3d, kde_3d # del event_dt_perm, gauss_3d, kde_3d
chunk_boot_KDE = chunk_permutation(select_event_dt, conv_tt, repetitions % chunk_size, max_jitter, kernal_w, kernal_h) chunk_boot_KDE = chunk_permutation(select_event_dt, conv_tt, repetitions % chunk_size, max_jitter, kernal_w, kernal_h)
chunk_collector.extend(chunk_boot_KDE) chunk_collector.extend(chunk_boot_KDE)
chunk_collector = np.array(chunk_collector)
# ToDo: this works but is incorrect i think
chunk_collector /= np.sum(chunk_collector, axis=1).reshape(chunk_collector.shape[0], 1)
print(f'bootstrap with {repetitions:.0f} repetitions took {time.time() - t0:.2f}s.') print(f'bootstrap with {repetitions:.0f} repetitions took {time.time() - t0:.2f}s.')
embed() # fig, ax = plt.subplots()
quit() # for i in range(len(chunk_collector)):
# ax.plot(cp.asnumpy(conv_t), chunk_collector[i])
return cp.asnumpy(conv_t), chunk_collector
def main(base_path): def main(base_path):
@ -202,10 +206,15 @@ def main(base_path):
lose_chrips_centered_on_ag_off_t.append(event_centered_times(ag_on_off_t_GRID[:, 1], chirp_times[1])) lose_chrips_centered_on_ag_off_t.append(event_centered_times(ag_on_off_t_GRID[:, 1], chirp_times[1]))
norm_count.append(len(chirp_times[1])) norm_count.append(len(chirp_times[1]))
kde(np.hstack(lose_chrips_centered_on_ag_off_t)) kde_array = kde(np.hstack(lose_chrips_centered_on_ag_off_t))
conv_t, boot_kde = permulation_kde(np.hstack(lose_chrips_centered_on_ag_off_t), norm_count=norm_count)
permulation_kde(np.hstack(lose_chrips_centered_on_ag_off_t), norm_count=norm_count) fig, ax = plt.subplots()
for i in range(len(boot_kde)):
ax.plot(conv_t, boot_kde[i])
ax.plot(conv_t, kde_array, color='k', lw=3)
pass pass
if __name__ == '__main__': if __name__ == '__main__':