Guest User

Untitled

a guest
Mar 22nd, 2018
64
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.36 KB | None | 0 0
  1. from sklearn.metrics.pairwise import euclidean_distances
  2.  
  3. distance_matrix = euclidean_distances(weights)
  4. print(distance_matrix.shape)
  5.  
  6. similar_words = {search_term: [id2word[idx] for idx in distance_matrix[word2id[search_term]-1].argsort()[1:6]+1]
  7. for search_term in ['god', 'jesus', 'noah', 'egypt', 'john', 'gospel', 'moses','famine']}
  8.  
  9. similar_words
Add Comment
Please, Sign In to add comment