loader_test.py Script with example for loading files from the dataset.

main_net.py Script to train and evaluate. Data Generator takes about 3h to generate. So for first run you should save the training and validation generators and load them in future runs.

net/ configurations, models, utils, generator + model save file

mlruns/ Automatically created by MlFlow to store metrics, etc.

classes/ annotation + defines classes for edf files and loading format

Conda env setup conda config --add channels conda-forge conda config --set channel_priority strict conda env create -n ENV_NAME -f environment.yml