Diff of /code/eda.py [000000] .. [365bd4]

Switch to side-by-side view

--- 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