/TAN

code for TAN model

Primary LanguagePython

Code For TAN (Temporal attention networks for biomedical hypothesis generation)

Temporal attention networks for biomedical hypothesis generation Accepted by The Journal of Biomedical informatics. link

Data

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

Dependencies

  • Ubuntu 20.04.2 LTS
  • RTX 2080
  • Python 3.8.5
  • Cuda 11.2

Environments

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

Run

Please enter bash Imm_train.sh or bash Vir_train.sh or bash Neu_train.sh at the terminal to run the program.