Natalie Koh, Zachary Laswick, Daiwei Shen
- STEP
- Graph WaveNet
- Simple graph convolutional network with LSTM layer implemented in Keras
- For data pre-processing, see
PruneDatasets_SingleSubject.ipynb
. - To run STEP on the datasets, use scripts in STEP/ModifiedSTEPCode.
- To run Graph WaveNET,
cd
into the WaveNet directory and runpython train.py --gcn_bool
. - To run the simple GCN implemented in Keras, use
KerasGNNwLSTM.ipynb
.