diff --git a/README.md b/README.md index 0072c39..c4ef9fc 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/code/utils.py b/code/utils.py index 1501203..2e4e79c 100644 --- a/code/utils.py +++ b/code/utils.py @@ -1,7 +1,7 @@ import h5py import numpy as np import random - +# WINDOW_SIZE = 100 def rescale_array(X): @@ -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)) \ No newline at end of file + return (seq[pos:pos + size] for pos in range(0, len(seq), size))