Skip to content

Commit a003c85

Browse files
Update demo.py
fix code
1 parent b3f516b commit a003c85

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

demo.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -20,17 +20,17 @@ def main():
2020

2121
if len(q_list) == 1:
2222
print("相似詞前 100 排序")
23-
res = model.most_similar(q_list[0],topn = 100)
23+
res = model.wv.most_similar(q_list[0],topn = 100)
2424
for item in res:
2525
print(item[0]+","+str(item[1]))
2626

2727
elif len(q_list) == 2:
2828
print("計算 Cosine 相似度")
29-
res = model.similarity(q_list[0],q_list[1])
29+
res = model.wv.similarity(q_list[0],q_list[1])
3030
print(res)
3131
else:
3232
print("%s之於%s,如%s之於" % (q_list[0],q_list[2],q_list[1]))
33-
res = model.most_similar([q_list[0],q_list[1]], [q_list[2]], topn= 100)
33+
res = model.wv.most_similar([q_list[0],q_list[1]], [q_list[2]], topn= 100)
3434
for item in res:
3535
print(item[0]+","+str(item[1]))
3636
print("----------------------------")

0 commit comments

Comments
 (0)