Dammio
Dammio

Reputation: 1059

How to find the most similar words from a set of input words by CBOW (GenSim)?

I use GenSim and CBOW for training the corpus. How can I get the most similar words from a set of input words?

For example: Given a set of input words: ["David", "Mary", "married"]. Can I infer some output words like: "wedding", "husband", "wife", "couple", etc?

Upvotes: 0

Views: 1652

Answers (1)

Stanislas Morbieu
Stanislas Morbieu

Reputation: 1827

You can use the wv.most_similar method of your model.

Upvotes: 2

Related Questions