摘要
本发明公开了一种基于异构注意力机制的草药与症状关系建模方法,包括以下步骤:基于图神经网络结构,构建异构注意力关系网络;所述异构注意力关系网络包括草药节点、症状节点和连接节点的边,所述连接节点的边类型包括草药‑草药边、草药‑症状边和症状‑症状边;基于连接节点的边类型,获取每个节点的注意力分数;基于每个节点的注意力分数,聚合同质节点和异质节点的信息,获得增强后的节点表示;基于增强后的节点表示,构建草药与症状之间的上下文依赖模型,进行草药推荐。本发明实现了基于异构注意力机制的草药与症状关系建模,为中药推荐系统提供了更准确、高效的特征表示方法。