基于注意力机制的多维特征融合的分子性质预测的模型
摘要:
本发明提供一种基于注意力机制的多维特征融合的分子性质预测的模型,所述模型包括分子1D序列特征向量生成模块、分子2D原子图特征向量生成模块、分子3D结构特征向量生成模块及多维特征融合模块,其中所述分子3D结构特征向量生成模块使用SphereNet作为特征解码器。基于注意力机制的多维特征编码器能够从多个维度处理药物分子的SMILES字符串,从不同的角度获取特征信息,使得编码信息更全面。使用多维度特征的融合能够进一步压缩药物特征向量,高质量的编码能够将特征与预测结果密切关联。
0/0