This repository contains the source code ,the data and trained models.

Dependencies:

• python 3.6
• pytorch >= 1.2.0
• numpy
• RDkit = 2019.03.3.0
• pandas
• Gensim >=3.4.0

The data sets with train/test splits are provided as .7z file in a directory called 'data'.

The test set is created specially for label reversal experiments.

1.mol_featurizer.py generates input for TransformerCPI model.

2.main.py trains TransformerCPI model.

Lifan Chen

Mingyue Zheng

Lifan Chen, Xiaoqin Tan, Dingyan Wang, Feisheng Zhong, Xiaohong Liu, Tianbiao Yang, Xiaomin Luo, Kaixian Chen, Hualiang Jiang, Mingyue Zheng, TransformerCPI: improving compound–protein interaction prediction by sequence-based deep learning with self-attention mechanism and label reversal experiments, Bioinformatics, Volume 36, Issue 16, 15 August 2020, Pages 4406–4414, https://doi.org/10.1093/bioinformatics/btaa524

TransformerCPI2.0 is now available at https://github.com/lifanchen-simm/transfomerCPI2.0 !