输出的权重矩阵如何得到？

[zaibian]

您好，side info的权重矩阵即embedding怎么得到呢？好像程序中没有这一块的逻辑

[NiDHanWang]

您好，side info的权重矩阵即embedding怎么得到呢？好像程序中没有这一块的逻辑

你好，请问这个问题解决了嘛，我也没找到

[jh-lau]

restore_var = [v for v in tf.compat.v1.trainable_variables() if v.name.find('embedding') != -1]
saver = tf.compat.v1.train.Saver()
saver.restore(sess, "checkpoints/EGES")
e1, e2, e3, e4 = sess.run(restore_var)

注意自行计算非oov item还需要结合attention_embedding。
计算oov item可以按照论文使用side info embedding的mean pooling。