def read_student(totalDB):
dsets = h5py.File(totalDB,'r')
dset = dsets['d100'] #학생당 실험당 하나 추출
dset = np.array(dset)
dset = dset[:,:,50:200] #실험전 -0.4 ~ -0.1초만 추출
dset = pp.filterByFrequency(dset,[2]) #alpha 채널만 분리
dset = dset[0,:,:,:] #filterByBrequncy 함수결과가 4차원으로 오기 떄문에 1차원 버림
return dset
def init():
totalDB = 'D:/workspace/eeg/minlab/test2.hdf5'
chnames= ['Fp1','Fp2','F7','F3','Fz','F4','F8','FC5','FC1','FC2','FC6','T7','C3','Cz','C4','T8','CP5','CP1','CP2','CP6','P7','P3','Pz','P4','P8','PO9','O1','Oz','O2','PO10']
dsetRaw = retreiveDset(totalDB)
dsetAbs = dectectAbsOutlier(dsetRaw)
dset, dsetGapOutliers = detectGapOutlier(dsetAbs)
dsetSmall = dset[:,:,50:200]
dsetAlpha = pp.filterByFrequency(dsetSmall,[2])
dsetAlpha = dsetAlpha[0]
dsetAlphaByChnTime = np.apply_along_axis(np.mean,1, dsetAlpha)
mneObj = createMneObject(dsetAlphaByChnTime,chnames)
clustRes = eabs.exe_kmeans(dsetAlphaByChnTime)
return dsetAlpha,dsetAlphaByChnTime, clustRes, mneObj, chnames
