Update README.md
Signed-off-by: David Rotermund <54365609+davrot@users.noreply.github.com>
This commit is contained in:
parent
255728b5ef
commit
633642c203
1 changed files with 4 additions and 3 deletions
|
@ -342,12 +342,12 @@ def calculate_spectral_coherence(
|
||||||
)
|
)
|
||||||
|
|
||||||
if trial_id == 0:
|
if trial_id == 0:
|
||||||
calculation = wave_data_a * wave_data_b
|
calculation = wave_data_a * np.conj(wave_data_b)
|
||||||
norm_data_a = np.abs(wave_data_a) ** 2
|
norm_data_a = np.abs(wave_data_a) ** 2
|
||||||
norm_data_b = np.abs(wave_data_b) ** 2
|
norm_data_b = np.abs(wave_data_b) ** 2
|
||||||
|
|
||||||
else:
|
else:
|
||||||
calculation += wave_data_a * wave_data_b
|
calculation += wave_data_a * np.conj(wave_data_b)
|
||||||
norm_data_a += np.abs(wave_data_a) ** 2
|
norm_data_a += np.abs(wave_data_a) ** 2
|
||||||
norm_data_b += np.abs(wave_data_b) ** 2
|
norm_data_b += np.abs(wave_data_b) ** 2
|
||||||
|
|
||||||
|
@ -355,7 +355,7 @@ def calculate_spectral_coherence(
|
||||||
norm_data_a /= float(n_trials)
|
norm_data_a /= float(n_trials)
|
||||||
norm_data_b /= float(n_trials)
|
norm_data_b /= float(n_trials)
|
||||||
|
|
||||||
coherence = np.abs(calculation) ** 2 / (norm_data_a * norm_data_b)
|
coherence = np.abs(calculation) ** 2 / ((norm_data_a * norm_data_b) + 1e-20)
|
||||||
|
|
||||||
return np.nanmean(coherence, axis=-1), frequency_axis, t
|
return np.nanmean(coherence, axis=-1), frequency_axis, t
|
||||||
|
|
||||||
|
@ -377,6 +377,7 @@ rng = np.random.default_rng(1)
|
||||||
n_t: int = 1000
|
n_t: int = 1000
|
||||||
n_trials: int = 100
|
n_trials: int = 100
|
||||||
t: np.ndarray = np.arange(0, n_t) * dt
|
t: np.ndarray = np.arange(0, n_t) * dt
|
||||||
|
|
||||||
amplitude: float = 0.75
|
amplitude: float = 0.75
|
||||||
|
|
||||||
x = dt * 2.0 * np.pi * (f_base + f_delta * 2 * (rng.random((n_t, n_trials)) - 0.5))
|
x = dt * 2.0 * np.pi * (f_base + f_delta * 2 * (rng.random((n_t, n_trials)) - 0.5))
|
||||||
|
|
Loading…
Reference in a new issue