EEG-Denoise-Seizure-Detection/commands.txt at master · Rohit-R-Rao/EEG-Denoise-Seizure-Detection · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
Preprocessing

Papers:

python 1_preprocess.py --data_type train --cpu_num 12 --label_type tse_bi --save_directory out/ --samplerate 200
python 1_preprocess.py --data_type dev --cpu_num 12 --label_type tse_bi --save_directory out/ --samplerate 200

Denoising:

python denoising.py --data_type train --cpu_num 12 --label_type tse_bi --save_directory out/ --samplerate 200
python denoising.py --data_type dev --cpu_num 12 --label_type tse_bi --save_directory out/ --samplerate 200

python denoising.py --save_directory out/ --data_type dev --label_type tse_bi


resnet_short_lstm_raw

1. resnet_short_lstm_raw

CUDA_VISIBLE_DEVICES=0 python3 ./2_train.py \
    --project-name resnet_short_lstm_raw \
    --model resnet_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --epoch 8 \
    --batch-size 32 \
    --seizure-wise-eval-for-binary True


2. # swt, bandpass, notch

CUDA_VISIBLE_DEVICES=0 python3 ./3_test.py \
    --project-name resnet_short_lstm_raw \
    --model resnet_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --seed 0 \
    --denoise bandpass \
    --ignore-model-speed \
    --cpu 1


3. swt, bandpass, notch

CUDA_VISIBLE_DEVICES=0 python3 ./4_seiz_test.py \
    --project-name resnet_short_lstm_raw \
    --model resnet_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --seizure-wise-eval-for-binary True \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --best \
    --seed 0 \
    --denoise bandpass


resnet_distilation_lstm

1.

CUDA_VISIBLE_DEVICES=0 python3 ./2_train.py \
    --project-name resnet_dilation_lstm_raw \
    --model resnet_dilation_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --epoch 8 \
    --batch-size 32 \
    --seizure-wise-eval-for-binary True


2. Testing swt, bandpass, notch

CUDA_VISIBLE_DEVICES=0 python3 ./3_test.py \
    --project-name resnet_dilation_lstm_raw \
    --model resnet_dilation_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --seed 0 \
    --denoise bandpass \
    --ignore-model-speed \
    --cpu 1


3. Seizure Test swt, bandpass, notch

CUDA_VISIBLE_DEVICES=0 python3 ./4_seiz_test.py \
    --project-name resnet_dilation_lstm_raw \
    --model resnet_dilation_lstm \
    --task-type binary \
    --optim adam \
    --window-size 4 \
    --window-shift 1 \
    --eeg-type bipolar \
    --enc-model raw \
    --binary-sampler-type 6types \
    --binary-target-groups 2 \
    --seizure-wise-eval-for-binary True \
    --best \
    --seed 0 \
    --denoise bandpass