摘要
本发明公开了一种基于最大团算法的中医药方推荐方法,根据从公开中医药数据集中获取的药方数据生成症状集合-中药集合,从而得到样本数据集,根据样本数据集构造症状-中药图并获取每味中药的正例中药集合和负例中药集合,构建包括图神经网络、症状表示向量筛选模块、多层感知机、中药表示向量拼接模块、中药预测模块和中药关系预测模块的中药预测模型-,并采用样本数据集进行训练,得到训练好的中药预测模型以及中药关系图,将需要进行中药药方推荐的症状集合输入训练好的中药预测模型得到推荐中药集合,最后根据推荐中药集合和中药关系图搜索得到最终的推荐药方。本发明可以使得生成的推荐药方更加准确并符合实际。
- 单位