From dbd542742fbaabb3d15add6dafaa9dacc4dda6cc Mon Sep 17 00:00:00 2001 From: Sherry-XLL Date: Mon, 4 Apr 2022 02:43:28 +0000 Subject: [PATCH 1/2] FIX: fix UserWarning in get_norm_adj_mat and accelerate --- recbole/model/general_recommender/lightgcn.py | 2 +- recbole/model/general_recommender/ngcf.py | 2 +- recbole/model/general_recommender/sgl.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/recbole/model/general_recommender/lightgcn.py b/recbole/model/general_recommender/lightgcn.py index 898b87011..78844ce34 100644 --- a/recbole/model/general_recommender/lightgcn.py +++ b/recbole/model/general_recommender/lightgcn.py @@ -100,7 +100,7 @@ def get_norm_adj_mat(self): L = sp.coo_matrix(L) row = L.row col = L.col - i = torch.LongTensor([row, col]) + i = torch.LongTensor(np.array([row, col])) data = torch.FloatTensor(L.data) SparseL = torch.sparse.FloatTensor(i, data, torch.Size(L.shape)) return SparseL diff --git a/recbole/model/general_recommender/ngcf.py b/recbole/model/general_recommender/ngcf.py index cb82d87da..0080f32eb 100644 --- a/recbole/model/general_recommender/ngcf.py +++ b/recbole/model/general_recommender/ngcf.py @@ -103,7 +103,7 @@ def get_norm_adj_mat(self): L = sp.coo_matrix(L) row = L.row col = L.col - i = torch.LongTensor([row, col]) + i = torch.LongTensor(np.array([row, col])) data = torch.FloatTensor(L.data) SparseL = torch.sparse.FloatTensor(i, data, torch.Size(L.shape)) return SparseL diff --git a/recbole/model/general_recommender/sgl.py b/recbole/model/general_recommender/sgl.py index d88aec04f..453fefc1c 100644 --- a/recbole/model/general_recommender/sgl.py +++ b/recbole/model/general_recommender/sgl.py @@ -156,7 +156,7 @@ def csr2tensor(self, matrix: sp.csr_matrix): """ matrix = matrix.tocoo() x = torch.sparse.FloatTensor( - torch.LongTensor([matrix.row.tolist(), matrix.col.tolist()]), + torch.LongTensor(np.array([matrix.row, matrix.col])), torch.FloatTensor(matrix.data.astype(np.float32)), matrix.shape ).to(self.device) return x From 390c305bb4c5d1969ef642a1b13f2910b7806e91 Mon Sep 17 00:00:00 2001 From: Sherry-XLL Date: Mon, 4 Apr 2022 02:58:12 +0000 Subject: [PATCH 2/2] FIX: fix UserWarning in get_norm_adj_mat --- recbole/model/general_recommender/ncl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recbole/model/general_recommender/ncl.py b/recbole/model/general_recommender/ncl.py index 5006b4c7f..1c78c5fcd 100644 --- a/recbole/model/general_recommender/ncl.py +++ b/recbole/model/general_recommender/ncl.py @@ -121,7 +121,7 @@ def get_norm_adj_mat(self): L = sp.coo_matrix(L) row = L.row col = L.col - i = torch.LongTensor([row, col]) + i = torch.LongTensor(np.array([row, col])) data = torch.FloatTensor(L.data) SparseL = torch.sparse.FloatTensor(i, data, torch.Size(L.shape)) return SparseL