From b0490cfd9df5f3b4d30b108620b26a499bce39ce Mon Sep 17 00:00:00 2001
From: Arthur Batel <arthur@dm2l.insa-lyon.fr>
Date: Mon, 30 Sep 2024 15:37:21 +0200
Subject: [PATCH] get user emb
---
cornac/models/gcmc/nn_modules.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/cornac/models/gcmc/nn_modules.py b/cornac/models/gcmc/nn_modules.py
index 78a26fe7..fcc74545 100644
--- a/cornac/models/gcmc/nn_modules.py
+++ b/cornac/models/gcmc/nn_modules.py
@@ -62,8 +62,8 @@ class NeuralNetwork(nn.Module):
ifeat : torch.Tensor
The input item feature
"""
- user_out, item_out = self.encoder(enc_graph, ufeat, ifeat)
- pred_ratings = self.decoder(dec_graph, user_out, item_out)
+ self.user_out, self.item_out = self.encoder(enc_graph, ufeat, ifeat)
+ pred_ratings = self.decoder(dec_graph, self.user_out, self.item_out)
return pred_ratings
--
GitLab