kuandeng / LightGCN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

About the layer combination coefficients

zxm97 opened this issue · comments

commented

In function '_create_lightgcn_embed', the code computes E(0), A_fold_hat * E(0), A_fold_hat^2 * E(0)...A_fold_hat^K * E(0), then concatenates them and uses tf.reduce_mean to get the final representations. Where are the layer combination coefficients 1 / (K + 1) ?

The result of tf.reduce_mean function is the average of the K+1 embeddings , which equals to the weighted sum of the K+1 embeddings with coefficients 1/(K+1).

commented

Sorry, I confused k with K