|
a |
|
b/README.md |
|
|
1 |
# SGCN-for-epilepsy-detection |
|
|
2 |
This repository holds the source code of SGCN model. |
|
|
3 |
The code consists of two parts: Frequency-domain_Complex_network and SGCN. |
|
|
4 |
|
|
|
5 |
Dataset |
|
|
6 |
|
|
|
7 |
Due to the size limitation of upload file, we provide website of the Bonn dataset:http://epileptologie-bonn.de/cms/front_content.php?idcat=193&lang=3&changelang=3. |
|
|
8 |
|
|
|
9 |
Frequency-domain_Complex_network |
|
|
10 |
|
|
|
11 |
This part contains three code files, LVG.m, LPHVG.m, and graph_representation.m. In LPVG and LPHVG, we set the variable L of limited number of penetrations. When you set L to 0, you can build VG and HVG. In graph_representation.m, you can generate the complex network training_data, train_label, testing_data, and test_label. |
|
|
12 |
|
|
|
13 |
SGCN |
|
|
14 |
|
|
|
15 |
This part contains a code file, SGCN.py. The main file is SGCN.py. To use this classifier, please generate the complex networks in advance, and make sure that you have pytorch, Python 3 and all the packages we have used installed. |
|
|
16 |
|
|
|
17 |
Next, please take the following two steps. |
|
|
18 |
|
|
|
19 |
Step 1. Change the path in line 81 of SGCN.py to the path of input data. |
|
|
20 |
|
|
|
21 |
Step 2. Run the command in your command council. |