实现药品推荐的模型训练方法、装置及计算计设备与流程

文档序号:20017220发布日期:2020-02-25 10:47阅读:来源:国知局

技术特征:

1.一种实现药品推荐的模型训练方法,其特征在于,包括:

获取病患的病患数据和用药数据,通过词嵌入转化处理,由所述病患数据和所述用药数据分别得到病患特征和药品特征;

基于病患与药品的使用关系,以病患特征作为病患节点,以药品特征作为药品节点,进行病患与药品之间的关系图构建;

根据所述关系图的邻接结构,将所述关系图中节点表示的特征转化为图信息特征;

将所述图信息特征输入模型的编解码网络进行损失值计算,当计算到的损失值最小化,所述模型收敛。

2.如权利要求1所述的方法,其特征在于,所述病患的病患数据包括所述病患的入院记录、病史、检验检查报告和诊断信息;

所述通过词嵌入转化处理,由所述病患数据和所述用药数据分别得到病患特征和药品特征,包括:

对所述病患的入院记录、病史、检验检查报告和诊断信息分别进行词嵌入转化处理,得到对应的词嵌入向量;

将所述病患的入院记录、病史、检验检查报告和诊断信息分别对应的词嵌入向量进行拼接,得到所述病患特征。

3.如权利要求1或2所述的方法,其特征在于,所述病患的用药数据包括所述病患基于所述病患数据所使用药品的适应症信息;

所述通过词嵌入转化处理,由所述病患数据和所述用药数据分别得到病患特征和药品特征,包括:

对所述药品的适应症信息进行词嵌入转化处理,得到所述药品特征。

4.如权利要求1所述的方法,其特征在于,所述基于病患与药品的使用关系,以病患特征作为病患节点,以药品特征作为药品节点,进行病患与药品之间的关系图构建,包括:

根据所述病患对应的病患标识创建病患节点,将病患特征存储至病患节点,并根据所述病患使用药品对应的药品标识创建药品节点,将药品特征存储至药品节点;

基于病患与药品的使用关系,在所述病患节点与所述药品节点之间构建一条路径,形成病患与药品之间的关系图。

5.如权利要求1所述的方法,其特征在于,所述根据所述关系图的邻接结构,将所述关系图中节点表示的特征转化为图信息特征,包括:

针对所述关系图中的每一个节点,基于所述关系图的邻接结构,确定该节点在所述关系图中的邻居节点;

获取随机初始化的权重矩阵和所述邻居节点表示的特征;

根据所述权重矩阵和所述邻居节点表示的特征,计算得到由该节点表示的特征转化的图信息特征。

6.如权利要求1所述的方法,其特征在于,所述编解码网络包括多个编码网络层和多个解码网络层;

所述将所述图信息特征输入模型的编解码网络进行损失值计算,包括:

将所述图信息特征输入所述编解码网络中的第一个编码网络层,进行隐藏单元向量计算,并基于计算得到的隐藏单元向量计算第一个编码网络层的推荐损失值;

将计算得到的隐藏单元向量输入所述编解码网络中的第一个解码网络层,进行还原特征计算,并基于计算得到的还原特征计算第一个解码网络层的还原损失值;

对所述编解码网络中的多个编码网络层和多个解码网络层进行遍历,根据各编码网络层的推荐损失值和各解码网络层的还原损失值计算所述模型的损失值;

如果所述模型的损失值未达到最小化,则通过该损失值的反向传播,更新所述模型的权重矩阵,以继续进行模型训练,直至所述模型收敛。

7.如权利要求6所述的方法,其特征在于,所述将所述图信息特征输入所述编解码网络中的第一个编码网络层,进行隐藏单元向量计算,并根据计算得到的隐藏单元向量计算第一个编码网络层的推荐损失值之前,所述将所述图信息特征输入模型的编解码网络进行损失值计算,还包括:

根据设定冷启动条件,判断是否需要对所述图信息特征进行隐藏;

如果是,则将所述图信息特征设置为零。

8.如权利要求1所述的方法,其特征在于,所述将所述图信息特征输入模型的编解码网络进行损失值计算,当计算到的损失值最小化,所述模型收敛之后,所述方法还包括:

基于收敛的模型,对待推荐病患的病患数据进行推荐药品预测;

所述基于收敛的模型,对待推荐病患的病患数据进行推荐药品预测,包括:

对所述待推荐病患的病患数据进行词嵌入转化处理,得到待推荐病患特征;

基于关系图,由所述待推荐病患特征转化为待推荐图信息特征;

将所述待推荐图信息特征输入收敛的模型中编解码网络进行隐藏单元向量计算;

对计算得到的隐藏单元向量进行线性变换,得到推荐药品特征,根据所述推荐药品特征进行推荐药品的推荐。

9.一种实现药品推荐的模型训练装置,其特征在于,包括:

特征获取模块,用于获取病患的病患数据和用药数据,通过词嵌入转化处理,由所述病患数据和所述用药数据分别得到病患特征和药品特征;

图构建模块,用于基于病患与药品的使用关系,以病患特征作为病患节点,以药品特征作为药品节点,进行病患与药品之间的关系图构建;

图特征获取模块,用于根据所述关系图的邻接结构,将所述关系图中节点所表示的特征转化为图信息特征;

损失值计算模块,用于将所述图信息特征输入模型的编解码网络进行损失值计算,当计算到的损失值最小化,所述模型收敛。

10.一种计算机设备,其特征在于,包括:

处理器;及

存储器,所述存储器上存储有计算机可读指令,所述计算机可读指令被所述处理器执行时实现如权利要求1至8中任一项所述的实现药品推荐的模型训练方法。

当前第2页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1