摘要

本发明公开了一种基于证候信息的图神经网络中药推荐方法,获取若干病症-中药数据并按照证候进行分类,划分得到各个症候分类的病症-中药数据集合,根据每个病症-中药数据集合构建病症-中药矩阵,构建包含所有病症和中药的图神经网络,基于所有证候分类的病症-中药矩阵进行传播,得到病症、中药的嵌入表示向量,构建并训练得到中药推荐模型;当需要针对某个疾病进行中药推荐时,首先确定该疾病所包含的病症集合,然后基于图神经网络获取病症集合中各个病症的嵌入表示向量,将嵌入表示向量输入训练好的中药推荐模型,根据模型输出的得分函数选择得分较高的中药作为推荐中药。本发明将证候信息融入到图神经网络中,提升中药推荐的准确度。