search

recommenders-team / recommenders

Best Practices on Recommendation Systems
https://recommenders-team.github.io/recommenders/intro.html
MIT License
19.12k stars 3.09k forks source link

Save LightGCN models #1753

Open jiankunwang6 opened 2 years ago

jiankunwang6 commented 2 years ago

Description

Hi! I trained a lightgcn models and set save_model= True.

But when I check my directory, I got error below. Can anyone help me address this? image

Other Comments

gro1m commented 2 years ago

@jiankunwang6 Maybe going through this article may help you: https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/. But there is a lot of data stored and the data is stored in TensorFlow 1 format, TensorFlow 2 format would be easier... Related to: https://github.com/microsoft/recommenders/issues/1806

RanjanRavi2398 commented 1 year ago

Hi @jiankunwang6 , if you trying to save trained model in TensorFlow. image I think, it might be help you