/JGRN

The code for "Learning Graph in Graph Convolutional Neural Networks for Robust Seizure Prediction"

Primary LanguagePython

JGRN

The code for "Learning Graph in Graph Convolutional Neural Networks for Robust Seizure Prediction"

Data Generation

Use lib.data.generate_power_spectral_density.py to generate the dataset. Dataset related parameters can be modified in configs.dataset.

Parameter Tuning

Modify the (hyper-)parameters in the configs.parameters

Run

After modifing the dataset related and model related (hyper-)parameters, run main.py to obtain the results.