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,
cdinto the WaveNet directory and runpython train.py --gcn_bool. - To run the simple GCN implemented in Keras, use
KerasGNNwLSTM.ipynb.