summaryrefslogtreecommitdiff
path: root/scripts/most_similar.py
blob: ad780a7577d4c3be6641bb15fdb9a89e346eb3b3 (plain)
1
2
3
4
5
6
7
8
9
10
11
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"]))