Switch to unified view

a b/notebooks/Deep_EEG_Muse.ipynb
1
{
2
  "nbformat": 4,
3
  "nbformat_minor": 0,
4
  "metadata": {
5
    "colab": {
6
      "name": "Deep_EEG_Muse.ipynb",
7
      "provenance": [],
8
      "collapsed_sections": [],
9
      "toc_visible": true,
10
      "include_colab_link": true
11
    },
12
    "kernelspec": {
13
      "name": "python3",
14
      "display_name": "Python 3"
15
    },
16
    "accelerator": "GPU"
17
  },
18
  "cells": [
19
    {
20
      "cell_type": "markdown",
21
      "metadata": {
22
        "id": "view-in-github",
23
        "colab_type": "text"
24
      },
25
      "source": [
26
        "<a href=\"https://colab.research.google.com/github/kylemath/DeepEEG/blob/master/notebooks/Deep_EEG_Muse.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
27
      ]
28
    },
29
    {
30
      "cell_type": "markdown",
31
      "metadata": {
32
        "id": "cH7KRd8ZZPMd",
33
        "colab_type": "text"
34
      },
35
      "source": [
36
        "## DeepEEG\n"
37
      ]
38
    },
39
    {
40
      "cell_type": "code",
41
      "metadata": {
42
        "id": "KjZu4dFMFHJV",
43
        "colab_type": "code",
44
        "colab": {
45
          "base_uri": "https://localhost:8080/",
46
          "height": 734
47
        },
48
        "outputId": "c6c06cec-2bcb-49b4-ddae-32e16b5da89d"
49
      },
50
      "source": [
51
        "!git clone https://github.com/kylemath/DeepEEG\n",
52
        "!chmod +x ./DeepEEG/install.sh\n",
53
        "%cd DeepEEG\n",
54
        "!./install.sh\n",
55
        "from utils import *\n",
56
        "%matplotlib inline"
57
      ],
58
      "execution_count": 3,
59
      "outputs": [
60
        {
61
          "output_type": "stream",
62
          "text": [
63
            "Cloning into 'DeepEEG'...\n",
64
            "remote: Enumerating objects: 13, done.\u001b[K\n",
65
            "remote: Counting objects: 100% (13/13), done.\u001b[K\n",
66
            "remote: Compressing objects: 100% (12/12), done.\u001b[K\n",
67
            "remote: Total 947 (delta 2), reused 5 (delta 1), pack-reused 934\u001b[K\n",
68
            "Receiving objects: 100% (947/947), 19.62 MiB | 8.43 MiB/s, done.\n",
69
            "Resolving deltas: 100% (489/489), done.\n",
70
            "/content/DeepEEG/DeepEEG\n",
71
            "Requirement already up-to-date: pip in /usr/local/lib/python3.6/dist-packages (20.2.2)\n",
72
            "Requirement already satisfied: absl-py==0.7.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 1)) (0.7.0)\n",
73
            "Requirement already satisfied: astor==0.7.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 2)) (0.7.1)\n",
74
            "Requirement already satisfied: cycler==0.10.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 3)) (0.10.0)\n",
75
            "Requirement already satisfied: gast==0.2.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 4)) (0.2.2)\n",
76
            "Requirement already satisfied: grpcio==1.18.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 5)) (1.18.0)\n",
77
            "Requirement already satisfied: h5py==2.9.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 6)) (2.9.0)\n",
78
            "Requirement already satisfied: Keras==2.2.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 7)) (2.2.4)\n",
79
            "Requirement already satisfied: Keras-Applications==1.0.7 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 8)) (1.0.7)\n",
80
            "Requirement already satisfied: Keras-Preprocessing==1.0.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 9)) (1.0.9)\n",
81
            "Requirement already satisfied: kiwisolver==1.0.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 10)) (1.0.1)\n",
82
            "Requirement already satisfied: Markdown==3.0.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 11)) (3.0.1)\n",
83
            "Requirement already satisfied: matplotlib==3.0.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 12)) (3.0.2)\n",
84
            "Requirement already satisfied: mne==0.17.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 13)) (0.17.0)\n",
85
            "Requirement already satisfied: numpy==1.16.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 14)) (1.16.1)\n",
86
            "Requirement already satisfied: pandas==0.24.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 15)) (0.24.1)\n",
87
            "Requirement already satisfied: protobuf==3.6.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 16)) (3.6.1)\n",
88
            "Requirement already satisfied: pyparsing==2.3.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 17)) (2.3.1)\n",
89
            "Requirement already satisfied: python-dateutil==2.8.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 18)) (2.8.0)\n",
90
            "Requirement already satisfied: pytz==2018.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 19)) (2018.9)\n",
91
            "Requirement already satisfied: PyYAML==4.2b1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 20)) (4.2b1)\n",
92
            "Requirement already satisfied: scikit-learn==0.20.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 21)) (0.20.2)\n",
93
            "Requirement already satisfied: scipy==1.2.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 22)) (1.2.0)\n",
94
            "Requirement already satisfied: six==1.12.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 23)) (1.12.0)\n",
95
            "Requirement already satisfied: sklearn==0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 24)) (0.0)\n",
96
            "Requirement already satisfied: tensorboard==1.12.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 25)) (1.12.2)\n",
97
            "Requirement already satisfied: tensorflow==1.13.0rc1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 26)) (1.13.0rc1)\n",
98
            "Requirement already satisfied: termcolor==1.1.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 27)) (1.1.0)\n",
99
            "Requirement already satisfied: Werkzeug==0.14.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 28)) (0.14.1)\n",
100
            "Requirement already satisfied: setuptools in /usr/local/lib/python3.6/dist-packages (from kiwisolver==1.0.1->-r requirements.txt (line 10)) (49.6.0)\n",
101
            "Requirement already satisfied: wheel>=0.26; python_version >= \"3\" in /usr/local/lib/python3.6/dist-packages (from tensorboard==1.12.2->-r requirements.txt (line 25)) (0.35.1)\n",
102
            "Requirement already satisfied: tensorflow-estimator<1.14.0rc0,>=1.13.0rc0 in /usr/local/lib/python3.6/dist-packages (from tensorflow==1.13.0rc1->-r requirements.txt (line 26)) (1.13.0)\n",
103
            "Requirement already satisfied: mock>=2.0.0 in /usr/local/lib/python3.6/dist-packages (from tensorflow-estimator<1.14.0rc0,>=1.13.0rc0->tensorflow==1.13.0rc1->-r requirements.txt (line 26)) (4.0.2)\n"
104
          ],
105
          "name": "stdout"
106
        }
107
      ]
108
    },
109
    {
110
      "cell_type": "markdown",
111
      "metadata": {
112
        "id": "FhkDPfIq1ewa",
113
        "colab_type": "text"
114
      },
115
      "source": [
116
        "#Load Data, Concatenate\n"
117
      ]
118
    },
119
    {
120
      "cell_type": "code",
121
      "metadata": {
122
        "id": "jhU8hlcg1e6T",
123
        "colab_type": "code",
124
        "colab": {
125
          "base_uri": "https://localhost:8080/",
126
          "height": 1000
127
        },
128
        "outputId": "f5548962-7f6e-4dbb-83f4-618cf9c36f49"
129
      },
130
      "source": [
131
        "!git clone https://github.com/kylemath/eeg-notebooks_v0.1/\n",
132
        "data_dir = 'visual/cueing'\n",
133
        "\n",
134
        "subs = [101, 102, 103, 104, 105, 106, 108, 109, 110, 111, 112,\n",
135
        "        202, 203, 204, 205, 207, 208, 209, 210, 211,\n",
136
        "        301, 302, 303, 304, 305, 306, 307, 308, 309,\n",
137
        "        1101, 1102, 1103, 1104, 1105, 1106, 1108, 1109, 1110,\n",
138
        "        1202, 1203, 1205, 1206, 1209, 1210, 1211, 1215,\n",
139
        "        1301, 1302, 1313, \n",
140
        "        1401, 1402, 1403, 1404, 1405,  1408, 1410, 1411, 1412, 1413, 1413, 1414, 1415, 1416]\n",
141
        "\n",
142
        "nsesh = 2\n",
143
        "event_id = {'LeftCue':1, 'RightCue':2}\n",
144
        "\n",
145
        "raw = LoadMuseData(subs,nsesh,data_dir)\n",
146
        "\n",
147
        "#subs = [ 1]\n",
148
        "#nsesh = 1\n",
149
        "#data_dir = 'visual/P300'\n",
150
        "#event_names = ['Standard','Target']\n",
151
        "\n",
152
        "#subs = [ 4]\n",
153
        "#nsesh = 1\n",
154
        "#data_dir = 'visual/SSVEP'\n",
155
        "#event_names = ['f30Hz','f20Hz']"
156
      ],
157
      "execution_count": 4,
158
      "outputs": [
159
        {
160
          "output_type": "stream",
161
          "text": [
162
            "Cloning into 'eeg-notebooks_v0.1'...\n",
163
            "remote: Enumerating objects: 2451, done.\u001b[K\n",
164
            "remote: Total 2451 (delta 0), reused 0 (delta 0), pack-reused 2451\u001b[K\n",
165
            "Receiving objects: 100% (2451/2451), 166.74 MiB | 12.34 MiB/s, done.\n",
166
            "Resolving deltas: 100% (1096/1096), done.\n",
167
            "Checking out files: 100% (632/632), done.\n",
168
            "Loading Data\n",
169
            "Subject number 1/63\n",
170
            " Session number 1/2\n",
171
            "No files for subject with filename []\n",
172
            " Session number 2/2\n",
173
            "No files for subject with filename []\n",
174
            "Subject number 2/63\n",
175
            " Session number 1/2\n",
176
            "No files for subject with filename []\n",
177
            " Session number 2/2\n",
178
            "No files for subject with filename []\n",
179
            "Subject number 3/63\n",
180
            " Session number 1/2\n",
181
            "No files for subject with filename []\n",
182
            " Session number 2/2\n",
183
            "No files for subject with filename []\n",
184
            "Subject number 4/63\n",
185
            " Session number 1/2\n",
186
            "No files for subject with filename []\n",
187
            " Session number 2/2\n",
188
            "No files for subject with filename []\n",
189
            "Subject number 5/63\n",
190
            " Session number 1/2\n",
191
            "No files for subject with filename []\n",
192
            " Session number 2/2\n",
193
            "No files for subject with filename []\n",
194
            "Subject number 6/63\n",
195
            " Session number 1/2\n",
196
            "No files for subject with filename []\n",
197
            " Session number 2/2\n",
198
            "No files for subject with filename []\n",
199
            "Subject number 7/63\n",
200
            " Session number 1/2\n",
201
            "No files for subject with filename []\n",
202
            " Session number 2/2\n",
203
            "No files for subject with filename []\n",
204
            "Subject number 8/63\n",
205
            " Session number 1/2\n",
206
            "No files for subject with filename []\n",
207
            " Session number 2/2\n",
208
            "No files for subject with filename []\n",
209
            "Subject number 9/63\n",
210
            " Session number 1/2\n",
211
            "No files for subject with filename []\n",
212
            " Session number 2/2\n",
213
            "No files for subject with filename []\n",
214
            "Subject number 10/63\n",
215
            " Session number 1/2\n",
216
            "No files for subject with filename []\n",
217
            " Session number 2/2\n",
218
            "No files for subject with filename []\n",
219
            "Subject number 11/63\n",
220
            " Session number 1/2\n",
221
            "No files for subject with filename []\n",
222
            " Session number 2/2\n",
223
            "No files for subject with filename []\n",
224
            "Subject number 12/63\n",
225
            " Session number 1/2\n",
226
            "No files for subject with filename []\n",
227
            " Session number 2/2\n",
228
            "No files for subject with filename []\n",
229
            "Subject number 13/63\n",
230
            " Session number 1/2\n",
231
            "No files for subject with filename []\n",
232
            " Session number 2/2\n",
233
            "No files for subject with filename []\n",
234
            "Subject number 14/63\n",
235
            " Session number 1/2\n",
236
            "No files for subject with filename []\n",
237
            " Session number 2/2\n",
238
            "No files for subject with filename []\n",
239
            "Subject number 15/63\n",
240
            " Session number 1/2\n",
241
            "No files for subject with filename []\n",
242
            " Session number 2/2\n",
243
            "No files for subject with filename []\n",
244
            "Subject number 16/63\n",
245
            " Session number 1/2\n",
246
            "No files for subject with filename []\n",
247
            " Session number 2/2\n",
248
            "No files for subject with filename []\n",
249
            "Subject number 17/63\n",
250
            " Session number 1/2\n",
251
            "No files for subject with filename []\n",
252
            " Session number 2/2\n",
253
            "No files for subject with filename []\n",
254
            "Subject number 18/63\n",
255
            " Session number 1/2\n",
256
            "No files for subject with filename []\n",
257
            " Session number 2/2\n",
258
            "No files for subject with filename []\n",
259
            "Subject number 19/63\n",
260
            " Session number 1/2\n",
261
            "No files for subject with filename []\n",
262
            " Session number 2/2\n",
263
            "No files for subject with filename []\n",
264
            "Subject number 20/63\n",
265
            " Session number 1/2\n",
266
            "No files for subject with filename []\n",
267
            " Session number 2/2\n",
268
            "No files for subject with filename []\n",
269
            "Subject number 21/63\n",
270
            " Session number 1/2\n",
271
            "No files for subject with filename []\n",
272
            " Session number 2/2\n",
273
            "No files for subject with filename []\n",
274
            "Subject number 22/63\n",
275
            " Session number 1/2\n",
276
            "No files for subject with filename []\n",
277
            " Session number 2/2\n",
278
            "No files for subject with filename []\n",
279
            "Subject number 23/63\n",
280
            " Session number 1/2\n",
281
            "No files for subject with filename []\n",
282
            " Session number 2/2\n",
283
            "No files for subject with filename []\n",
284
            "Subject number 24/63\n",
285
            " Session number 1/2\n",
286
            "No files for subject with filename []\n",
287
            " Session number 2/2\n",
288
            "No files for subject with filename []\n",
289
            "Subject number 25/63\n",
290
            " Session number 1/2\n",
291
            "No files for subject with filename []\n",
292
            " Session number 2/2\n",
293
            "No files for subject with filename []\n",
294
            "Subject number 26/63\n",
295
            " Session number 1/2\n",
296
            "No files for subject with filename []\n",
297
            " Session number 2/2\n",
298
            "No files for subject with filename []\n",
299
            "Subject number 27/63\n",
300
            " Session number 1/2\n",
301
            "No files for subject with filename []\n",
302
            " Session number 2/2\n",
303
            "No files for subject with filename []\n",
304
            "Subject number 28/63\n",
305
            " Session number 1/2\n",
306
            "No files for subject with filename []\n",
307
            " Session number 2/2\n",
308
            "No files for subject with filename []\n",
309
            "Subject number 29/63\n",
310
            " Session number 1/2\n",
311
            "No files for subject with filename []\n",
312
            " Session number 2/2\n",
313
            "No files for subject with filename []\n",
314
            "Subject number 30/63\n",
315
            " Session number 1/2\n",
316
            "No files for subject with filename []\n",
317
            " Session number 2/2\n",
318
            "No files for subject with filename []\n",
319
            "Subject number 31/63\n",
320
            " Session number 1/2\n",
321
            "No files for subject with filename []\n",
322
            " Session number 2/2\n",
323
            "No files for subject with filename []\n",
324
            "Subject number 32/63\n",
325
            " Session number 1/2\n",
326
            "No files for subject with filename []\n",
327
            " Session number 2/2\n",
328
            "No files for subject with filename []\n",
329
            "Subject number 33/63\n",
330
            " Session number 1/2\n",
331
            "No files for subject with filename []\n",
332
            " Session number 2/2\n",
333
            "No files for subject with filename []\n",
334
            "Subject number 34/63\n",
335
            " Session number 1/2\n",
336
            "No files for subject with filename []\n",
337
            " Session number 2/2\n",
338
            "No files for subject with filename []\n",
339
            "Subject number 35/63\n",
340
            " Session number 1/2\n",
341
            "No files for subject with filename []\n",
342
            " Session number 2/2\n",
343
            "No files for subject with filename []\n",
344
            "Subject number 36/63\n",
345
            " Session number 1/2\n",
346
            "No files for subject with filename []\n",
347
            " Session number 2/2\n",
348
            "No files for subject with filename []\n",
349
            "Subject number 37/63\n",
350
            " Session number 1/2\n",
351
            "No files for subject with filename []\n",
352
            " Session number 2/2\n",
353
            "No files for subject with filename []\n",
354
            "Subject number 38/63\n",
355
            " Session number 1/2\n",
356
            "No files for subject with filename []\n",
357
            " Session number 2/2\n",
358
            "No files for subject with filename []\n",
359
            "Subject number 39/63\n",
360
            " Session number 1/2\n",
361
            "No files for subject with filename []\n",
362
            " Session number 2/2\n",
363
            "No files for subject with filename []\n",
364
            "Subject number 40/63\n",
365
            " Session number 1/2\n",
366
            "No files for subject with filename []\n",
367
            " Session number 2/2\n",
368
            "No files for subject with filename []\n",
369
            "Subject number 41/63\n",
370
            " Session number 1/2\n",
371
            "No files for subject with filename []\n",
372
            " Session number 2/2\n",
373
            "No files for subject with filename []\n",
374
            "Subject number 42/63\n",
375
            " Session number 1/2\n",
376
            "No files for subject with filename []\n",
377
            " Session number 2/2\n",
378
            "No files for subject with filename []\n",
379
            "Subject number 43/63\n",
380
            " Session number 1/2\n",
381
            "No files for subject with filename []\n",
382
            " Session number 2/2\n",
383
            "No files for subject with filename []\n",
384
            "Subject number 44/63\n",
385
            " Session number 1/2\n",
386
            "No files for subject with filename []\n",
387
            " Session number 2/2\n",
388
            "No files for subject with filename []\n",
389
            "Subject number 45/63\n",
390
            " Session number 1/2\n",
391
            "No files for subject with filename []\n",
392
            " Session number 2/2\n",
393
            "No files for subject with filename []\n",
394
            "Subject number 46/63\n",
395
            " Session number 1/2\n",
396
            "No files for subject with filename []\n",
397
            " Session number 2/2\n",
398
            "No files for subject with filename []\n",
399
            "Subject number 47/63\n",
400
            " Session number 1/2\n",
401
            "No files for subject with filename []\n",
402
            " Session number 2/2\n",
403
            "No files for subject with filename []\n",
404
            "Subject number 48/63\n",
405
            " Session number 1/2\n",
406
            "No files for subject with filename []\n",
407
            " Session number 2/2\n",
408
            "No files for subject with filename []\n",
409
            "Subject number 49/63\n",
410
            " Session number 1/2\n",
411
            "No files for subject with filename []\n",
412
            " Session number 2/2\n",
413
            "No files for subject with filename []\n",
414
            "Subject number 50/63\n",
415
            " Session number 1/2\n",
416
            "No files for subject with filename []\n",
417
            " Session number 2/2\n",
418
            "No files for subject with filename []\n",
419
            "Subject number 51/63\n",
420
            " Session number 1/2\n",
421
            "No files for subject with filename []\n",
422
            " Session number 2/2\n",
423
            "No files for subject with filename []\n",
424
            "Subject number 52/63\n",
425
            " Session number 1/2\n",
426
            "No files for subject with filename []\n",
427
            " Session number 2/2\n",
428
            "No files for subject with filename []\n",
429
            "Subject number 53/63\n",
430
            " Session number 1/2\n",
431
            "No files for subject with filename []\n",
432
            " Session number 2/2\n",
433
            "No files for subject with filename []\n",
434
            "Subject number 54/63\n",
435
            " Session number 1/2\n",
436
            "No files for subject with filename []\n",
437
            " Session number 2/2\n",
438
            "No files for subject with filename []\n",
439
            "Subject number 55/63\n",
440
            " Session number 1/2\n",
441
            "No files for subject with filename []\n",
442
            " Session number 2/2\n",
443
            "No files for subject with filename []\n",
444
            "Subject number 56/63\n",
445
            " Session number 1/2\n",
446
            "No files for subject with filename []\n",
447
            " Session number 2/2\n",
448
            "No files for subject with filename []\n",
449
            "Subject number 57/63\n",
450
            " Session number 1/2\n",
451
            "No files for subject with filename []\n",
452
            " Session number 2/2\n",
453
            "No files for subject with filename []\n",
454
            "Subject number 58/63\n",
455
            " Session number 1/2\n",
456
            "No files for subject with filename []\n",
457
            " Session number 2/2\n",
458
            "No files for subject with filename []\n",
459
            "Subject number 59/63\n",
460
            " Session number 1/2\n",
461
            "No files for subject with filename []\n",
462
            " Session number 2/2\n",
463
            "No files for subject with filename []\n",
464
            "Subject number 60/63\n",
465
            " Session number 1/2\n",
466
            "No files for subject with filename []\n",
467
            " Session number 2/2\n",
468
            "No files for subject with filename []\n",
469
            "Subject number 61/63\n",
470
            " Session number 1/2\n",
471
            "No files for subject with filename []\n",
472
            " Session number 2/2\n",
473
            "No files for subject with filename []\n",
474
            "Subject number 62/63\n",
475
            " Session number 1/2\n",
476
            "No files for subject with filename []\n",
477
            " Session number 2/2\n",
478
            "No files for subject with filename []\n",
479
            "Subject number 63/63\n",
480
            " Session number 1/2\n",
481
            "No files for subject with filename []\n",
482
            " Session number 2/2\n",
483
            "No files for subject with filename []\n"
484
          ],
485
          "name": "stdout"
486
        },
487
        {
488
          "output_type": "error",
489
          "ename": "TypeError",
490
          "evalue": "ignored",
491
          "traceback": [
492
            "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
493
            "\u001b[0;31mTypeError\u001b[0m                                 Traceback (most recent call last)",
494
            "\u001b[0;32m<ipython-input-4-6099eec806a7>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[1;32m     13\u001b[0m \u001b[0mevent_id\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m{\u001b[0m\u001b[0;34m'LeftCue'\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;36m1\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'RightCue'\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;36m2\u001b[0m\u001b[0;34m}\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     14\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 15\u001b[0;31m \u001b[0mraw\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mLoadMuseData\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0msubs\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0mnsesh\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0mdata_dir\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m     16\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m     17\u001b[0m \u001b[0;31m#subs = [ 1]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
495
            "\u001b[0;32m/content/DeepEEG/utils.py\u001b[0m in \u001b[0;36mLoadMuseData\u001b[0;34m(subs, nsesh, data_dir, load_verbose, sfreq)\u001b[0m\n\u001b[1;32m    106\u001b[0m       raw.append(muse_load_data(data_dir, sfreq=sfreq ,subject_nb=sub,\n\u001b[1;32m    107\u001b[0m                     session_nb=isesh+1,verbose=load_verbose))\n\u001b[0;32m--> 108\u001b[0;31m   \u001b[0mraw\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mconcatenate_raws\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mraw\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    109\u001b[0m   \u001b[0;32mreturn\u001b[0m \u001b[0mraw\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    110\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
496
            "\u001b[0;32m/usr/local/lib/python3.6/dist-packages/mne/io/base.py\u001b[0m in \u001b[0;36mconcatenate_raws\u001b[0;34m(raws, preload, events_list, verbose)\u001b[0m\n",
497
            "\u001b[0;32m/usr/local/lib/python3.6/dist-packages/mne/utils.py\u001b[0m in \u001b[0;36mverbose\u001b[0;34m(function, *args, **kwargs)\u001b[0m\n\u001b[1;32m    950\u001b[0m         \u001b[0;32mwith\u001b[0m \u001b[0muse_log_level\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mverbose_level\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    951\u001b[0m             \u001b[0;32mreturn\u001b[0m \u001b[0mfunction\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m*\u001b[0m\u001b[0margs\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 952\u001b[0;31m     \u001b[0;32mreturn\u001b[0m \u001b[0mfunction\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m*\u001b[0m\u001b[0margs\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m**\u001b[0m\u001b[0mkwargs\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    953\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    954\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
498
            "\u001b[0;32m/usr/local/lib/python3.6/dist-packages/mne/io/base.py\u001b[0m in \u001b[0;36mconcatenate_raws\u001b[0;34m(raws, preload, events_list, verbose)\u001b[0m\n\u001b[1;32m   2635\u001b[0m         \u001b[0mfirst\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlast\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mzip\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m*\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mr\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfirst_samp\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mr\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mlast_samp\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;32mfor\u001b[0m \u001b[0mr\u001b[0m \u001b[0;32min\u001b[0m \u001b[0mraws\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m   2636\u001b[0m         \u001b[0mevents\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mconcatenate_events\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mevents_list\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfirst\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mlast\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m-> 2637\u001b[0;31m     \u001b[0mraws\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;36m0\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mappend\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mraws\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;36m1\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mpreload\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m   2638\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m   2639\u001b[0m     \u001b[0;32mif\u001b[0m \u001b[0mevents_list\u001b[0m \u001b[0;32mis\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
499
            "\u001b[0;31mTypeError\u001b[0m: append() takes exactly one argument (2 given)"
500
          ]
501
        }
502
      ]
503
    },
504
    {
505
      "cell_type": "markdown",
506
      "metadata": {
507
        "id": "oEd72sIHuiFc",
508
        "colab_type": "text"
509
      },
510
      "source": [
511
        "#Run Preprocessing\n",
512
        "\n",
513
        "**Input:  mne.raw, event_id**\n",
514
        "\n",
515
        "**Output: mne.epochs**\n",
516
        "```python\n",
517
        "plot_psd=False\n",
518
        "filter_data=True\n",
519
        "eeg_filter_highpass=1\n",
520
        "plot_events=False\n",
521
        "epoch_time=(-.2,1)\n",
522
        "baseline=(-.2,0)\n",
523
        "rej_thresh_uV=200\n",
524
        "rereference=False\n",
525
        "emcp_raw=False\n",
526
        "emcp_epochs=False\n",
527
        "epoch_decim=1\n",
528
        "plot_electrodes=False\n",
529
        "plot_erp=False\n",
530
        "```"
531
      ]
532
    },
533
    {
534
      "cell_type": "code",
535
      "metadata": {
536
        "id": "3HiHs8CluhjP",
537
        "colab_type": "code",
538
        "colab": {
539
          "base_uri": "https://localhost:8080/",
540
          "height": 85
541
        },
542
        "outputId": "70c40caf-689c-4f7e-868d-7b94affceeb7"
543
      },
544
      "source": [
545
        "epochs = PreProcess(raw,event_id)\n"
546
      ],
547
      "execution_count": null,
548
      "outputs": [
549
        {
550
          "output_type": "stream",
551
          "text": [
552
            "Filtering Data Between 1 and 30 Hz.\n",
553
            "4583 events found\n",
554
            "Event IDs: [ 1  2 11 12 21 22]\n",
555
            "Remaining Trials: 1408\n"
556
          ],
557
          "name": "stdout"
558
        }
559
      ]
560
    },
561
    {
562
      "cell_type": "markdown",
563
      "metadata": {
564
        "id": "pwqD_voiDypa",
565
        "colab_type": "text"
566
      },
567
      "source": [
568
        "#Run FeatureEngineer\n",
569
        "**Input:  mne.epochs**\n",
570
        "\n",
571
        "**Output: deepeeg.feats**\n",
572
        "```python\n",
573
        "model_type='NN'\n",
574
        "frequency_domain=False\n",
575
        "normalization=True\n",
576
        "electrode_median=False\n",
577
        "wavelet_decim=1\n",
578
        "flims=(3,30)\n",
579
        "f_bins=20\n",
580
        "wave_cycles=3\n",
581
        "spect_baseline=[-1,-.5]\n",
582
        "electrodes_out=[11,12,13,14,15]\n",
583
        "test_split = 0.2\n",
584
        "val_split = 0.2\n",
585
        "random_seed=1017\n",
586
        "watermark = False\n",
587
        "```"
588
      ]
589
    },
590
    {
591
      "cell_type": "code",
592
      "metadata": {
593
        "id": "i_jYlTW1A6sb",
594
        "colab_type": "code",
595
        "colab": {
596
          "base_uri": "https://localhost:8080/",
597
          "height": 221
598
        },
599
        "outputId": "4a93d84c-7ea9-45a9-d312-f801fb29fcb1"
600
      },
601
      "source": [
602
        "feats = FeatureEngineer(epochs,model_type='NN')"
603
      ],
604
      "execution_count": null,
605
      "outputs": [
606
        {
607
          "output_type": "stream",
608
          "text": [
609
            "Constructing Time Domain Features\n",
610
            "Normalizing X\n",
611
            "Combined X Shape: (1408, 256, 4)\n",
612
            "Combined Y Shape: (1408,)\n",
613
            "Y Example (should be 1s & 0s): [1 1 0 0 0 0 0 1 0 0]\n",
614
            "X Range: -12.106191:9.781672\n",
615
            "Input Shape: (256, 4)\n",
616
            "x_train shape: (844, 256, 4)\n",
617
            "844 train samples\n",
618
            "282 test samples\n",
619
            "282 validation samples\n",
620
            "Class Weights: [0.97011494 1.03178484]\n"
621
          ],
622
          "name": "stdout"
623
        }
624
      ]
625
    },
626
    {
627
      "cell_type": "markdown",
628
      "metadata": {
629
        "id": "9ChMvr-jKp8P",
630
        "colab_type": "text"
631
      },
632
      "source": [
633
        "# Run CreateModel\n",
634
        "\n",
635
        "**Input: deepeeg.feats**\n",
636
        "\n",
637
        "**Output: deepeeg.model, deepeeg.encoder**\n",
638
        "\n",
639
        "```python\n",
640
        "units=[16,8,4,8,16]\n",
641
        "dropout=.25\n",
642
        "batch_norm=True\n",
643
        "filt_size=3\n",
644
        "pool_size=2\n",
645
        "```"
646
      ]
647
    },
648
    {
649
      "cell_type": "code",
650
      "metadata": {
651
        "id": "HFf3rBbJKqHR",
652
        "colab_type": "code",
653
        "colab": {}
654
      },
655
      "source": [
656
        "model, _ = CreateModel(feats)"
657
      ],
658
      "execution_count": null,
659
      "outputs": []
660
    },
661
    {
662
      "cell_type": "markdown",
663
      "metadata": {
664
        "id": "-EKBjSapfE4O",
665
        "colab_type": "text"
666
      },
667
      "source": [
668
        "# TrainTestVal\n",
669
        "\n",
670
        "**Input: deepEEG.model, deepEEG.feats**\n",
671
        "\n",
672
        "```python\n",
673
        "batch_size=2\n",
674
        "train_epochs=20\n",
675
        "show_plots=True\n",
676
        "```"
677
      ]
678
    },
679
    {
680
      "cell_type": "code",
681
      "metadata": {
682
        "id": "u6ize7eJfB3J",
683
        "colab_type": "code",
684
        "colab": {}
685
      },
686
      "source": [
687
        "TrainTestVal(model, feats)"
688
      ],
689
      "execution_count": null,
690
      "outputs": []
691
    }
692
  ]
693
}