--- a +++ b/code/eda.py @@ -0,0 +1,28 @@ +import os +import h5py +import numpy as np +import matplotlib.pyplot as plt +import datetime as dt +import collections +import librosa + +path = "/media/ml/data_ml/EEG/deepsleepnet/data_npy/SC4061E0.npz" + +data = np.load(path) + +x = data['x'] +y = data['y'] + +fig_1 = plt.figure(figsize=(12, 6)) +plt.plot(x[100, ...].ravel()) +plt.title("EEG Epoch") +plt.ylabel("Amplitude") +plt.xlabel("Time") +plt.show() + +fig_2 = plt.figure(figsize=(12, 6)) +plt.plot(y.ravel()) +plt.title("Sleep Stages") +plt.ylabel("Classes") +plt.xlabel("Time") +plt.show() \ No newline at end of file