diff options
Diffstat (limited to 'scripts/most_similar.py')
| -rw-r--r-- | scripts/most_similar.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/scripts/most_similar.py b/scripts/most_similar.py new file mode 100644 index 0000000..ad780a7 --- /dev/null +++ b/scripts/most_similar.py @@ -0,0 +1,12 @@ +#!/usr/bin/python3 + +import gensim + +import config + +# load model +model = gensim.models.Word2Vec.load(config.model) +wv = model.wv +del model + +print(wv.most_similar_cosmul(positive=["lol"])) |