ranker

japanese/ranker.py

@@ -1,19 +1,3 @@
-import numpy
-import pickle
-
-import numpy as np
-import os
-import re
-import string
-import tojson
-from multiprocessing import Pool
-
-import sys
-import tojson
-import re
-import string
-
-
 import numpy as np
 from numpy.linalg import norm
 
@@ -144,4 +128,4 @@ class DirectedCentralityRnak(object):
             similarity_matrix.append(self.get_similarity_matrix(embedded))
         for matrix, doc_vector, candidate_phrases_embeddings in zip(similarity_matrix, self.doc_embeddings, self.tokens_embeddings):
             extracted_list.append(self._rank_part(matrix, doc_vector, candidate_phrases_embeddings))
-        return extracted_list
+        return extracted_list