update gitignore

Author: Captain

.gitignore

@@ -2,4 +2,6 @@
 data/
 .idea/
 result/
-venv/
+venv/
+.ipynb_checkpoints/
+*.ipynb

data_manager.py

@@ -59,6 +59,7 @@ def read_rating(self, path):
         return results
 
     # 生成词向量矩阵
+    # 这部分可以用torchwordemb代替
     def read_pretrained_word2vec(self, path, vocab, dim):
         parent_path = '/'.join(path.split('/')[:-1]) + '/'
         if os.path.isfile(parent_path + 'preW.all'):

models.py

@@ -31,7 +31,6 @@ def ConvMF(res_dir, train_user, train_item, valid_user, test_user,
     Test_R = test_user[1]
     Valid_R = valid_user[1]
 
-    # 这部分暂时不知道有什么用
     if give_item_weight is True:
         item_weight = np.array([math.sqrt(len(i))
                                 for i in Train_R_J], dtype=float)

text_analysis/cnn_model.py

@@ -19,7 +19,9 @@ def __init__(self, output_dimesion, vocab_size, dropout_rate, emb_dim, max_len,
         projection_dimension = output_dimesion
         self.qual_conv_set = {}
 
-        self.embedding = nn.Embedding()
+        '''Embedding Layer'''
+        if init_W is None:
+            self.embedding = nn.Embedding(max_features, emb_dim)
 
         self.conv1 = nn.Sequential(
             nn.Conv2d(1, nb_filters, kernel_size=(3, emb_dim)),