Merge pull request #2 from thinkall/master

Fix dim err of candidate_news_vector in prediction

src/model/DKN/__init__.py

@@ -97,7 +97,7 @@ def get_prediction(self, candidate_news_vector, clicked_news_vector):
         """
         # candidate_size, len(window_sizes) * num_filters
         user_vector = self.attention(candidate_news_vector,
-                                     clicked_news_vector.unsqueeze(dim=0))
+                                     clicked_news_vector.expand(candidate_news_vector.size(0), -1, -1))
         # candidate_size
         click_probability = self.click_predictor(candidate_news_vector,
                                                  user_vector)