摘要
本申请公开了一种基于改进图注意力机制的未知物毒性预测方法和装置,通过获取提取的未知混合物,采用SMILES码将分子片段封装为第一DGL图对象,输入至预训练的毒性预测模型中,输出第一毒性预测结果;将分子片段混合物与数据库中的已知分子进行指纹匹配,选取最佳匹配分子,将最佳匹配分子SMILES码封装为第二DGL图对象,并利用预构建的毒性预测模型对第二DGL图对象进行毒性预测,输出第二毒性预测结果,本申请通过对图注意力网络改进,增加点积注意力,对多层图注意力网络进行优化,提升特征提取能力和泛化能力,拥有更好的预测精度。
技术关键词
毒性预测方法
注意力机制
分子
多任务学习模型
混合物
对象
指纹
网络
特征提取能力
全局平均池化
线性
训练集
预测装置
节点
特征值
表达式
模块
参数
数据