Skip to content
Snippets Groups Projects
Unverified Commit 91ed98c6 authored by Theis Jendal's avatar Theis Jendal Committed by GitHub
Browse files

Fix layer combination coefficients (#606)

parent fa26b733
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
cornac/models/lightgcn/lightgcn.py
+ 0
− 3
View file @ 91ed98c6
...@@ -96,9 +96,6 @@ class Model(nn.Module): ...@@ -96,9 +96,6 @@ class Model(nn.Module):
item_embeds = h_dict[ITEM_KEY] item_embeds = h_dict[ITEM_KEY]
iterator = enumerate(zip([in_g] * len(self.layers), self.layers)) iterator = enumerate(zip([in_g] * len(self.layers), self.layers))
user_embeds = user_embeds * (1 / (len(self.layers) + 1))
item_embeds = item_embeds * (1 / (len(self.layers) + 1))
for k, (g, layer) in iterator: for k, (g, layer) in iterator:
h_dict = layer(g, h_dict) h_dict = layer(g, h_dict)
ue = h_dict[USER_KEY] ue = h_dict[USER_KEY]
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment