20 lines
678 B
Python
20 lines
678 B
Python
from utils.file_utils import load_preprocessed_data, get_epochs
|
|
|
|
|
|
def check_peaks():
|
|
"""
|
|
Sanity check for the "get_peaks" method
|
|
"""
|
|
import matplotlib.pyplot as plt
|
|
raw = load_preprocessed_data('002', 'N170')
|
|
epochs, _ = get_epochs(raw, [('face', 'intact')], picks='P7')
|
|
ch, latency, peak = epochs.average().get_peak(tmin=0.13, tmax=0.2, mode='neg', return_amplitude=True)
|
|
import numpy as np
|
|
plt.plot(epochs.times, np.squeeze(epochs.average().data.T))
|
|
plt.vlines([0.13, 0.2], -0.00001, 0.00001, colors='r', linestyles='dotted')
|
|
plt.vlines(latency, -0.00001, 0.00001, colors='gray', linestyles='dotted')
|
|
plt.show()
|
|
|
|
|
|
check_peaks()
|