一种基于医学知识图谱对比学习的药物预测方法

本发明公开了一种基于医学知识图谱对比学习的药物预测方法，将医疗数据集中的药物作为点，药物之间的关系作为边，构建多关系医学知识图谱；根据药物间的关系类型构建无向同质网络，节点预处理后确定训练集和测试集，对训练集和测试集中节点对关系设置标签；关系子图预处理后，判断关系子图的连通性；引入图神经网络和对比学习，构建多关系医学知识图谱对比学习模型，确定网络中节点的嵌入表示和每个关系类型对应关系的关系子图的嵌入表示；通过多关系医学知识图谱对比学习模型得到节点的嵌入表示，将其以串联的方式输入到机器学习分类器中，得到药物相互作用的预测结果。