命令model.most_similar(positive=['france'], topn=100)会返回与单词“France”最接近的前100个单词。但是,我想知道是否有一种方法可以输出相似度高于给定单词的相似度阈值的最相似单词。是否有以下方法?: model.most_similar(positive=['france'], threshold=0.9)
好的,假设你可以。请尝试以下代码:
def find_most_similar(model, wrd, threshold=0.75):
res = [item for item in model.wv.most_similar(wrd, topn=len(model.wv.vocab)) if item[1] > threshold]
return res