/NSSINet

Multi-Level Generative Adversarial Network Framework for Non-Suicidal Self-Injury Detection through High-Dimensional EEG Signals

Primary LanguagePython

NSSINet

Multi-Level Generative Adversarial Network Framework for Non-Suicidal Self-Injury Detection through High-Dimensional EEG Signals

Installation

  • Python 3.7
  • Pytorch 1.3.1
  • NVIDIA CUDA 9.2
  • Numpy 1.20.3
  • Scikit-learn 0.23.2
  • scipy 1.3.1

Training

  • NSSINet model definition file: model_deap_autoencoder.py
  • Pipeline of the NSSINet: train_autoencoder.py
  • Implementation of domain adversarial training: adv_layer.py

Usage

  • After modify setting (path, etc), just run the main function in the train_autoencoder.py