recommenders-team/recommenders

[ASK] How can I save SASRec model for re-training and prediction?

Open

#1,718 创建于 2022年5月13日

在 GitHub 查看
 (2 评论) (0 反应) (0 负责人)Python (2,972 fork)batch import
help wanted

仓库指标

Star
 (17,706 star)
PR 合并指标
 (平均合并 6天 16小时) (30 天内合并 10 个 PR)

描述

I have tried to save trained SASRec model.

pickle, tf.saved_model.save, model.save(), and surprise.dump are not working.

While saving, I got warning saying 'Found untraced functions', and while loading, 'AttributeError: 'SASREC' object has no attribute 'seq_max_len''.

Plz someone let me know how to save and load SASRec model!

贡献者指南