训练关系网络嵌入模型和确定使用概率的方法及装置

摘要

本说明书实施例提供一种训练关系网络嵌入模型和确定使用概率的方法及装置。方法包括：基于关系网络中节点特征、边特征和第一参数集，确定各个节点的初级迭代的节点嵌入向量，以及各个连接边的初级迭代的边嵌入向量；执行多级向量迭代，以确定各个连接边的多级迭代的边嵌入向量，每级向量迭代包括，对于各个连接边，至少基于第二参数集，确定该连接边的本级迭代的边嵌入向量；对于各个第一类连接边，基于该连接边的多级迭代的边嵌入向量和预测参数集，确定该连接边的预测值；调整第一参数集、第二参数集和预测参数集中的参数值，使得预定义的损失函数达到最小，其中损失函数基于各个第一类连接边的预测值和标签值而确定。能够提高确定准确性。