| 12345678910111213141516171819 |
- from agent.core.es import bm25_vector_search
- from agent.core.vector import get_embeddings
- phrase = "渔业产品批发,海带边周转"
- phrase_vector = get_embeddings([phrase])[0]
- r = bm25_vector_search(phrase,phrase_vector)
- print(r)
- # from openpyxl import load_workbook
- # workbook = load_workbook('data/样本数据标注后-完整版.xlsx', data_only=True)
- # sheet = workbook['核对结果']
- # for row in sheet.iter_rows(min_row=2,min_col=2, max_col=3, values_only=True):
- # touxiang = row[0] # B列的值
- # yongtu = row[1] # C列的值
- # phrase = f"投向:{touxiang},用途:{yongtu}"
- # phrase_vector = get_embeddings([phrase])[0]
- # r = bm25_vector_search(phrase,phrase_vector)
- # print(f"输入:{phrase}")
- # print(f"输出:{r}")
- # workbook.close()
|
