科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于改进Bert4Rec的中医药方推荐方法,根据从公开中医药数据集中获取的药方数据生成症状集合-中药集合,从而得到样本数据集,对每个样本数据集分别进行掩码处理,然后对症状和中药分别进行编码得到编码向量,基于Bert4Rec模型构建包括输入处理模块,输入嵌入模块,Transformer模块,中药掩码特征向量筛选模块,中药预测模块和药方预测模块的药方推荐模型,采用掩码处理后的样本数据集对药方推荐模型进行训练,对于需要进行中药药方推荐的症状集合,生成全掩码的药方集合,一起输入训练好的药方预测模型得到推荐中药集合作为推荐药方。本发明基于改进Bert4Rec模型得到中医诊疗中涉及到的症状和中药两类实体的表示,实现更为准确的药方推荐。
- 单位
