recommenders-team/recommenders

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

Open

#1,718 opened on May 13, 2022

View on GitHubContributor guide
 (2 comments) (0 reactions) (0 assignees)Python (3,319 forks)batch import
help wanted

Repository metrics

Stars
 (21,866 stars)
PR merge metrics
 (PR metrics pending)

Description

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!

Contributor guide

Research direction
Review the SASRec model class definition and its serialization methods. Check if there is a custom save/load function or if the model inherits from Keras or TensorFlow. Also, check the repository's documentation or examples for model persistence.
Tech stack
python
Domain
aimachine learning
Issue type
Documentation
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-3 hours
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
PythonTensorFlow
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
70