智源發(fā)布新版bge詞嵌入模型
可在線運(yùn)行的notebook:在kaggle網(wǎng)站搜索
BAAI/bge-small-zh-v1.5
首先安裝所需依賴:
接著對(duì)模型進(jìn)行推理:
最終推理得到的文本相似度較為合理

模型推理有時(shí)候可能不能符合我們的業(yè)務(wù)場景,那么可以對(duì)模型進(jìn)行微調(diào)。我們首先準(zhǔn)確用于微調(diào)的數(shù)據(jù):
接著即可對(duì)模型進(jìn)行微調(diào):
完成模型微調(diào)后,我們加載新模型進(jìn)行測試:
最終我們得到的文本相似度為0.7,較為合理
標(biāo)簽: