Skip to content
Snippets Groups Projects
Commit b0490cfd authored by Arthur Batel's avatar Arthur Batel
Browse files

get user emb

parent 2c250995
Branches master
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
cornac/models/gcmc/nn_modules.py
+ 2
− 2
View file @ b0490cfd
...@@ -62,8 +62,8 @@ class NeuralNetwork(nn.Module): ...@@ -62,8 +62,8 @@ class NeuralNetwork(nn.Module):
ifeat : torch.Tensor ifeat : torch.Tensor
The input item feature The input item feature
""" """
user_out, item_out = self.encoder(enc_graph, ufeat, ifeat) self.user_out, self.item_out = self.encoder(enc_graph, ufeat, ifeat)
pred_ratings = self.decoder(dec_graph, user_out, item_out) pred_ratings = self.decoder(dec_graph, self.user_out, self.item_out)
return pred_ratings return pred_ratings
......
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