/TDE

code for TDE model.

Primary LanguagePython

Code For TDE Model

Data

The link of all three datasets is here. Please download and unzip them to data folder.

Dependencies

  • Ubuntu 20.04.2 LTS
  • RTX 3090
  • Python 3.6.13
  • Cuda 11.1

Environments

Please install dependency packages by pip install -r requirements.txt.

Run

Please enter bash pipeline_train_Imm.sh or bash pipeline_train_Vir.sh or bash pipeline_train_Neu.sh at the terminal to run the program.