alibaba/EasyRec

dssm模型训练好了后,如何得到user和item的向量。文档里面只有一个pai命令。

lzwscu opened this issue · 2 comments

我用easyRec里面的dssm模型,训练好了模型,然后也切割出了user模型和item模型。但是如何得到user向量和item向量呢?有详细的代码示例吗? 文档里面只有一个pai命令。我是直接从git上clone这个项目,然后安装的。没有用阿里云平台。
https://easyrec.readthedocs.io/en/latest/export.html

model split后分别对user模型和item模型执行predict操作。如果没有用使用maxcompute,可以做local prediction, 参考https://easyrec.readthedocs.io/en/latest/predict/Local%20%E7%A6%BB%E7%BA%BF%E9%A2%84%E6%B5%8B.html

参考 pai designer里面的案例代码:https://help.aliyun.com/zh/pai/use-cases/dssm-vector-recall