Ejemplo n.º 1
0
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")
Ejemplo n.º 2
0
 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
Ejemplo n.º 3
0
 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
Ejemplo n.º 4
0
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')