chelscelis
commited on
Commit
•
fc797e5
1
Parent(s):
493bb0a
Update utils.py
Browse files
utils.py
CHANGED
@@ -297,7 +297,7 @@ def rankResumes(text, df):
|
|
297 |
progressBar.progress(38, text = "Creating a Similarity Index...")
|
298 |
words = [word for word, count in dictionary.most_common()]
|
299 |
wordVectors = model.vectors_for_all(words, allow_inference = False)
|
300 |
-
indexer = AnnoyIndexer(wordVectors, num_trees =
|
301 |
similarityIndex = WordEmbeddingSimilarityIndex(wordVectors, kwargs = {'indexer': indexer})
|
302 |
progressBar.progress(50, text = "Creating a Similarity Matrix...")
|
303 |
similarityMatrix = SparseTermSimilarityMatrix(similarityIndex, dictionary, tfidf)
|
|
|
297 |
progressBar.progress(38, text = "Creating a Similarity Index...")
|
298 |
words = [word for word, count in dictionary.most_common()]
|
299 |
wordVectors = model.vectors_for_all(words, allow_inference = False)
|
300 |
+
indexer = AnnoyIndexer(wordVectors, num_trees = 300)
|
301 |
similarityIndex = WordEmbeddingSimilarityIndex(wordVectors, kwargs = {'indexer': indexer})
|
302 |
progressBar.progress(50, text = "Creating a Similarity Matrix...")
|
303 |
similarityMatrix = SparseTermSimilarityMatrix(similarityIndex, dictionary, tfidf)
|