import numpy as np

from glob import glob

import os

from sklearn.model_selection import train_test_split

base_path = "/media/ml/data_ml/EEG/deepsleepnet/data_npy"

files = glob(os.path.join(base_path, "*.npz"))

train_val, test = train_test_split(files, test_size=0.15, random_state=1337)

train, val = train_test_split(train_val, test_size=0.1, random_state=1337)

train_dict = {k: np.load(k) for k in train}

test_dict = {k: np.load(k) for k in test}

val_dict = {k: np.load(k) for k in val}