Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

# EEG_classification
Description of the approach : https://towardsdatascience.com/sleep-stage-classification-from-single-channel-eeg-using-convolutional-neural-networks-5c710d92d38e

Testing 101

Sleep Stage Classification from Single Channel EEG using Convolutional Neural
Networks
Expand Down
4 changes: 2 additions & 2 deletions code/utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import h5py
import numpy as np
import random

#
WINDOW_SIZE = 100

def rescale_array(X):
Expand Down Expand Up @@ -41,4 +41,4 @@ def gen(dict_files, aug=False):


def chunker(seq, size=WINDOW_SIZE):
return (seq[pos:pos + size] for pos in range(0, len(seq), size))
return (seq[pos:pos + size] for pos in range(0, len(seq), size))