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

changed example punit

Browse Source
This commit is contained in:
Jan Benda
2025-05-20 00:12:23 +02:00
parent 2e3d373100
commit ee2b8f98b7
18 changed files with 232 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
regimes.py
View File

@@ -3,10 +3,11 @@ import matplotlib.pyplot as plt
from pathlib import Path
from scipy.stats import linregress
from numba import jit
from thunderlab.tabledata import TableData
from plotstyle import plot_style, lighter, darker
model_cell = '2018-05-08-ad-invivo-1' # 228Hz, CV=0.67
data_path = Path('data')
sims_path = data_path / 'simulations'
@@ -200,14 +201,13 @@ def plot_psd(ax, s, path, contrast, spikes, nfft, dt, beatf1, beatf2):
def plot_example(axs, axr, axp, s, path, cell, alpha, beatf1, beatf2,
nfft, trials):
sim_path = path / f'{cell_name}-contrastspectrum-{1000*alpha:03.0f}.npz'
dt = 0.0001
tmax = nfft*dt
t1 = 0.1
spikes = punit_spikes(cell, alpha, beatf1, beatf2, tmax, trials)
plot_am(axs, s, alpha, beatf1, beatf2, t1)
plot_raster(axr, s, spikes, t1)
plot_psd(axp, s, sim_path, alpha, spikes, nfft, dt, beatf1, beatf2)
plot_psd(axp, s, path, alpha, spikes, nfft, dt, beatf1, beatf2)
def peak_ampl(freqs, psd, f):
@@ -285,17 +285,16 @@ def plot_peaks(ax, s, alphas, contrasts, powerf1, powerf2, powerfsum,
if __name__ == '__main__':
cell_name = '2018-05-08-ad-invivo-1' # 228Hz, CV=0.67
ratebase, cvbase, beatf1, beatf2, \
contrasts, powerf1, powerf2, powerfsum, powerfdiff = \
load_data(sims_path / f'{cell_name}-contrastpeaks.npz')
load_data(sims_path / f'{model_cell}-contrastpeaks.npz')
alphas = [0.002, 0.01, 0.03, 0.06]
parameters = load_models(data_path / 'punitmodels.csv')
cell = cell_parameters(parameters, cell_name)
cell = cell_parameters(parameters, model_cell)
nfft = 2**18
print(f'Loaded data for cell {cell_name}: '
print(f'Loaded data for cell {model_cell}: '
f'baseline rate = {ratebase:.0f}Hz, CV = {cvbase:.2f}')
s = plot_style()
@@ -308,7 +307,8 @@ if __name__ == '__main__':
# example power spectra:
for c, alpha in enumerate(alphas):
plot_example(axe[0, c], axe[1, c], axe[2, c], s, sims_path,
path = sims_path / f'{model_cell}-contrastspectrum-{1000*alpha:03.0f}.npz'
plot_example(axe[0, c], axe[1, c], axe[2, c], s, path,
cell, alpha, beatf1, beatf2, nfft, 100)
axe[1, 0].xscalebar(1, -0.1, 20, 'ms', ha='right')
axe[2, 0].legend(loc='center left', bbox_to_anchor=(0, -0.8),
@@ -322,3 +322,4 @@ if __name__ == '__main__':
powerfsum, powerfdiff)
fig.tag(axa, yoffs=2)
fig.savefig()
print()