from utils import *

# raw_filename = 'SimulatedERP_raw.fif'
# try:
# 	raw = mne.io.read_raw_fif(raw_filename, preload=True)
# 	event_id = {'CondZero':1, 'CondOne':2}
# except FileNotFoundError:
# 	#if not make the raw simulation and save
# 	raw,event_id = SimulateRaw(amp1=10, amp2=5, freq=2., batch=1)
# 	raw.save(raw_filename,overwrite=True)

raw,event_id = SimulateRaw(amp1=100, amp2=50, freq=2., batch=1)
epochs = PreProcess(raw,event_id,filter_data=False,plot_erp=True)

pick = 33
for event in event_id.keys():
	fig = plt.imshow(epochs[event]._data[:,pick,:])
	plt.show()

feats = FeatureEngineer(epochs)
model,_ = CreateModel(feats)
TrainTestVal(model,feats)