Skip to content

Instantly share code, notes, and snippets.

@mlai-demo
Last active April 18, 2020 19:13
Show Gist options
  • Save mlai-demo/8c04f06acf4f4d37d03a958729a2669a to your computer and use it in GitHub Desktop.
Save mlai-demo/8c04f06acf4f4d37d03a958729a2669a to your computer and use it in GitHub Desktop.
model_skipgram = gensim.models.Word2Vec(data, min_count = 1, size = 100, window = 5, sg = 1)
similar_words_skipgram = {search_term: [item for item in model_skipgram.wv.most_similar
([search_term], topn=300)]
for search_term in ['comorbidity']}
similar_words_skipgram
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment