Multi-Level Generative Adversarial Network Framework for Non-Suicidal Self-Injury Detection through High-Dimensional EEG Signals
- Python 3.7
- Pytorch 1.3.1
- NVIDIA CUDA 9.2
- Numpy 1.20.3
- Scikit-learn 0.23.2
- scipy 1.3.1
- NSSINet model definition file: model_deap_autoencoder.py
- Pipeline of the NSSINet: train_autoencoder.py
- Implementation of domain adversarial training: adv_layer.py
- After modify setting (path, etc), just run the main function in the train_autoencoder.py