import golem, psychic
from matplotlib import pyplot as plt
from annotate import annotate_horiz
trials = golem.DataSet.load("../../data/priming-trials.dat")
f = plt.figure(figsize=(12, 4))
f.add_axes([0.05, 0.2, 0.38, 0.7])
trials2 = psychic.nodes.Baseline([0.2, 1]).train_apply(trials, trials)
psychic.plot_erp(trials2.lix[["P3"], :, :], fig=f, pval=0, vspace=15)
annotate_horiz(-0.19, -0.01, 5, "baseline period", 0.5)
plt.gca().legend().set_visible(False)
plt.title("Uncorrected")
f.add_axes([0.55, 0.2, 0.38, 0.7])
trials2 = psychic.nodes.Baseline([-0.2, 0]).train_apply(trials, trials)
psychic.plot_erp(trials2.lix[["P3"], :, :], fig=f, pval=0, vspace=15)
annotate_horiz(-0.19, -0.01, 5, "baseline period", 0.5)
# plt.legend(loc='lower right')
plt.gca().legend().set_visible(False)
plt.title("Baseline corrected")
plt.savefig("baseline.png")
def _generate_debug_image(self, d):
""" Generate image describing the training data. """
fig = psychic.plot_erp(d)
fig.set_size_inches(7,11)
return fig
def _generate_debug_image(self, d):
""" Generate image describing the training data. """
d = psychic.DataSet(cl_lab=self.cl_lab, default=d)
fig = psychic.plot_erp(d, enforce_equal_n=False)
fig.set_size_inches(7, 11)
return fig
import psychic
import golem
from matplotlib import pyplot as plt
trials = golem.DataSet.load(psychic.find_data_path('priming-trials.dat'))
trials = psychic.nodes.Baseline((-0.2, 0)).train_apply(trials, trials)
psychic.plot_erp(trials.lix[['Fz', 'Cz', 'Pz'], :, :], fwer=None)
plt.savefig('plot_erp.png')
erp = psychic.nodes.ERP().apply(trials)
psychic.plot_erp(erp.lix[['Fz', 'Cz', 'Pz'], :, :])
plt.savefig('plot_erp_erp.png')