买专利卖专利找龙图腾,真高效! 查专利查商标用IPTOP,全免费!专利年费监控用IP管家,真方便!
申请/专利权人:小红书科技有限公司
摘要:本申请公开了训练图预训练模型的方法、对图进行分类的方法、装置。训练图预训练模型的方法包括:获取步骤,获取多个图的多个向量表示,每个向量表示包括多个节点和多个边;掩码步骤,利用多个任务,分别对多个向量表示进行掩码处理,得到与多个任务分别对应的多组掩码向量表示;构建步骤,利用多组掩码向量表示,构建与多个任务分别对应的多个损失函数;训练步骤,利用多个损失函数对图预训练模型进行训练,以得到训练后的图预训练模型。本发明的训练方法极大的提升了图预训练模型的性能。
主权项:1.一种训练图预训练模型的方法,用于电子设备,其特征在于,所述方法包括:获取步骤,获取多个图的多个向量表示,每个向量表示包括多个节点和多个边;掩码步骤,利用多个任务,分别对所述多个向量表示进行掩码处理,得到与所述多个任务分别对应的多组掩码向量表示;构建步骤,利用所述多组掩码向量表示,构建与所述多个任务分别对应的多个损失函数;训练步骤,利用多个所述损失函数对所述图预训练模型进行训练,以得到训练后的图预训练模型。
全文数据:
权利要求:
百度查询: 小红书科技有限公司 训练图预训练模型的方法、对图进行分类的方法和装置
免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。