def __generate_single_trial(self, label_nr, samples=1200):
t0 = 0.5*samples
rec = Recording(self.number_of_electrodes, samples)
for i in range(self.number_of_electrodes):
for t in range(samples):
fraction_of_signal = 1.0/(1+ math.exp(-(t-t0)))
rec.set_data(i, t, (1.0-fraction_of_signal)*np.random.normal(0.0, 0.1) + fraction_of_signal*(np.random.normal(0.0, 0.2*(label_nr+1)) + 0.1*math.cos(t*label_nr)))
rec.set_label(t, label_nr if t >= t0 else 0)
return rec
def load_recordings(self):
try:
data = np.loadtxt( self.filename )
except IOError as e:
print "I/O error({0}): {1}".format(e.errno, e.strerror)
samples = data.size/3 # a single recording
# samples = round(samples*0.03) # SPEEDUP HACK
nr_electrodes = data[0].size - 1 # first column is labels
rec = Recording(nr_electrodes, samples)
sample_index = 0
for t_vec in data[:samples]:
rec.set_label(sample_index, round(t_vec[0]))
for e in range(nr_electrodes):
rec.set_data(e, sample_index, t_vec[1+e])
sample_index += 1
return [rec]
