benda/nonlinearbaseline2025
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added correlation with baseline rate

Browse Source
This commit is contained in:
Jan Benda
2025-05-26 18:11:08 +02:00
parent 04578fd77e
commit d56c0c7c7e
4 changed files with 67 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
modelsusceptlown.py
View File

@@ -14,7 +14,7 @@ data_path = Path('data')
sims_path = data_path / 'simulations'
def plot_chi2_contrasts(axs, s, cell_name, nsegs=None):
def plot_chi2_contrasts(axs, s, cell_name, nsegs=None, vmax=None):
d = sims_path / f'{cell_name}-baseline.npz'
data = np.load(d)
rate = float(data['rate'])
@@ -23,14 +23,18 @@ def plot_chi2_contrasts(axs, s, cell_name, nsegs=None):
freqs, chi2, fcutoff, contrast, n = load_chi2(sims_path, cell_name,
None, nsegs)
ns = f'$N={n}$' if n < 1000 else f'$N=10^{np.log10(n):.0f}$'
cax = plot_chi2(axs[0], s, freqs, chi2, fcutoff, rate)
cax = plot_chi2(axs[0], s, freqs, chi2, fcutoff, rate, vmax)
cax.set_ylabel('')
axs[0].set_title(f'$c$=0\\,\\%, {ns}', fontsize='medium')
for k, alpha in enumerate([0.01, 0.03, 0.1]):
freqs, chi2, fcutoff, contrast, n = load_chi2(sims_path, cell_name,
alpha, nsegs)
ns = f'$N={n}$' if n < 1000 else f'$N=10^{np.log10(n):.0f}$'
cax = plot_chi2(axs[k + 1], s, freqs, chi2, fcutoff, rate)
cax = plot_chi2(axs[k + 1], s, freqs, chi2, fcutoff, rate, vmax)
if n < 1000:
vmax /= 10
else:
vmax /= 4
if alpha < 0.1:
cax.set_ylabel('')
axs[k + 1].set_title(f'$c$={100*alpha:g}\\,\\%, {ns}',
@@ -139,8 +143,8 @@ if __name__ == '__main__':
for ax in axs.flat:
ax.set_visible(False)
print('Example cells:')
plot_chi2_contrasts(axs[0], s, model_cell)
plot_chi2_contrasts(axs[1], s, model_cell, nsmall)
plot_chi2_contrasts(axs[0], s, model_cell, None, 40)
plot_chi2_contrasts(axs[1], s, model_cell, nsmall, 600)
for k in range(2):
fig.common_yticks(axs[k, :])
for k in range(4):