Merge pull request #1225 from Sherry-XLL/master

FIX: fix UserWarning in get_norm_adj_mat and accelerate csr2tensor

Author: 2017pxy

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

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

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

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