Python TUHEEGAbnormalDataset 예제들

예제 #1

파일: test_tuh_eeg_abnormal.py 프로젝트: diveu/PyEEGLab

 def test_dataset(self):
     dataset = TUHEEGAbnormalDataset(self.PATH)
     preprocessing = Pipeline([
         CommonChannelSet(),
         LowestFrequency(),
         BandPassFrequency(0.1, 47),
         ToDataframe(),
         DynamicWindow(4),
         JoinedPreprocessor(inputs=[[
             BinarizedSpearmanCorrelation(),
             CorrelationToAdjacency()
         ],
                                    Bandpower()],
                            output=GraphWithFeatures())
     ])
     dataset = dataset.set_pipeline(preprocessing).load()

예제 #2

파일: example_cnn_dense_classification_with_filter.py 프로젝트: rauswarn/PyEEGLab

from tensorflow.keras import Model, Input
from tensorflow.keras.layers import Dense, Concatenate, Reshape, Flatten, Conv2D, MaxPool2D
from tensorflow.keras.utils import to_categorical
from sklearn.metrics import accuracy_score, confusion_matrix
from sklearn.model_selection import StratifiedKFold
import numpy as np

import os
import sys

sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
from pyeeglab import    TUHEEGAbnormalDataset, SinglePickleCache, Pipeline, CommonChannelSet, \
                        LowestFrequency, BandPassFrequency, ToDataframe, DynamicWindow, \
                        BinarizedSpearmanCorrelation, ToNumpy

dataset = TUHEEGAbnormalDataset('../../data/tuh_eeg_abnormal/v2.0.0/edf')
dataset.set_cache_manager(SinglePickleCache('../../export'))

preprocessing = Pipeline([
    CommonChannelSet(),
    LowestFrequency(),
    BandPassFrequency(0.1, 47),
    ToDataframe(),
    DynamicWindow(8),
    BinarizedSpearmanCorrelation(),
    ToNumpy()
])

dataset = dataset.set_pipeline(preprocessing).load()
data, labels = dataset['data'], dataset['labels']

예제 #3

파일: example_cnn_dense_classification.py 프로젝트: xkazm/PyEEGLab

from tensorflow.keras.layers import Dense, Concatenate, Reshape, Flatten, Conv2D, MaxPool2D
from tensorflow.keras.utils import to_categorical
from sklearn.metrics import accuracy_score, confusion_matrix
from sklearn.model_selection import StratifiedKFold
import numpy as np

import os
import sys

sys.path.insert(
    0, os.path.abspath(os.path.join(os.path.dirname(__file__), '../..')))
from pyeeglab import    TUHEEGAbnormalDataset, Pipeline, CommonChannelSet, \
                        LowestFrequency, ToDataframe, DynamicWindow, BinarizedSpearmanCorrelation, \
                        ToNumpy

dataset = TUHEEGAbnormalDataset('../../data/tuh_eeg_abnormal/')

preprocessing = Pipeline([
    CommonChannelSet(),
    LowestFrequency(),
    ToDataframe(),
    DynamicWindow(8),
    BinarizedSpearmanCorrelation(),
    ToNumpy()
])

dataset = dataset.set_pipeline(preprocessing).load()
data, labels = dataset['data'], dataset['labels']

adjs = data[0].shape[0]
classes = len(set(labels))