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