开发者社区 > 大数据与机器学习 > 人工智能平台PAI > 正文

机器学习PAI easyrec 中的dssm模型保存的时候,可以分塔保存的吗?

机器学习PAI easyrec 中的dssm模型保存的时候,可以分塔保存的吗?直接export的时候,我发现只保存了一个完整的模型,那在线上预估的时候,分别生成 user embedding 和item embedding 不是有问题啊?

展开
收起
真的很搞笑 2024-06-18 12:54:05 54 0
2 条回答
写回答
取消 提交回答
  • 可以split成2个塔,https://easyrec.readthedocs.io/en/latest/export.html#id3 参考一下这个模型切分的文档 ,此回答整理自钉群“【EasyRec】推荐算法交流群”

    2024-06-18 19:51:18
    赞同 展开评论 打赏
  • 在EasyRec中,DSSM模型导出时,默认是将整个模型作为一个整体进行保存的,这通常包括了user tower和item tower的参数。您提到的希望分别保存user embedding和item embedding的需求,虽然直接通过export命令得到的是整合模型,但实际上在线上预测阶段,EasyRec框架设计了机制来支持从完整模型中分别提取user和item的embedding,可参考模型预测

    2024-06-18 16:43:04
    赞同 展开评论 打赏

人工智能平台 PAI(Platform for AI,原机器学习平台PAI)是面向开发者和企业的机器学习/深度学习工程平台,提供包含数据标注、模型构建、模型训练、模型部署、推理优化在内的AI开发全链路服务,内置140+种优化算法,具备丰富的行业场景插件,为用户提供低门槛、高性能的云原生AI工程化能力。

相关产品

  • 人工智能平台 PAI
  • 热门讨论

    热门文章

    相关电子书

    更多
    大规模机器学习在蚂蚁+阿里的应用 立即下载
    阿里巴巴机器学习平台AI 立即下载
    微博机器学习平台架构和实践 立即下载