一种基于自注意力机制与多药物特征组合的预测方法

摘要

本发明公开了一种基于自注意力机制与多药物特征组合的预测方法，包括药物分子经过扩展连通性指纹和Mol2Vec向量编译两种嵌入特征，由双向门控循环单元和邻域卷积抽取药物特征；所述药物中的蛋白质序列在经过特征嵌入后，利用一维卷积提取蛋白特征并与所述药物特征做相关注意力增强；拼接所述药物特征和所述蛋白特征，利用自注意力机制增强对蛋白质药物相互作用信息的提取；将所述拼接后的特征放入双向门控循环单元并预测蛋白质和药物的相互作用。将Morgan指纹编译和Mol2vec向量嵌入结合，使所提取的药物特征信息更为丰富；发现卷积网络结合门控循环单元提取蛋白质和药物特征，配合注意力机制强化提取蛋白质和药物之间的关系特征，可有效提升模型的性能。