预估生成user_embedding时，提取YoutubeDNN模型结构报错

shenweichen / DeepMatch

A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.

https://deepmatch.readthedocs.io/en/latest/

Apache License 2.0

2.22k stars 530 forks source link

预估生成user_embedding时，提取YoutubeDNN模型结构报错 #104

Open Ringokohi123 opened 4 months ago

Ringokohi123 commented 4 months ago

Describe the bug 成功加载训练好的YoutubeDNN模型后，使用Model(input=model.user_input,outputs=model.user_embedding)时，报Attributeerror：model has no attribute 'user_input'。请大神们指点迷津！

Operating environment

python version [3.6]
tensorflow version [1.14.0]
deepmatch version [0.2.0]

Ringokohi123 commented 4 months ago

如果在同一个脚本中，执行完模型训练，紧接着预估user_embedding不会报错。但是当训练和预估拆成两个脚本，预估时先load_model，再提取模型结构生成user_embedding则会报上述错误。考虑可能是模型中某些自定义的内容没有保存下来，但是还没有找到解决方法