科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于中药关系预测的图神经网络药方推荐方法,根据从公开中医药数据集中获取的药方数据生成症状集合-中药集合,从而得到样本数据集,根据样本数据集构造症状-中药图并获取每味中药的正例中药集合和负例中药集合,构建基于图神经网络的中医实体表示模型、药方推荐模型和中药关系预测模型,将这三个模型组合成一个多任务网络并采用样本数据集进行训练,得到训练好的中医实体表示模型和药方推荐模型,将需要进行中药药方推荐的症状集合输入训练好的药方推荐模型得到推荐中药集合。本发明将中药关系预测作为药方推荐的辅助任务进行多任务联合训练,在显式挖掘中药关系、增强中药实体表示的同时,实现较为准确的药方推荐。
- 单位
