dsProject/dsRag/T2_Txt2Vec.py

'''
# 安装向量化的包 （# 断开VPN后执行安装包）
conda activate rag
pip install text2vec torch torchvision torchaudio

# 安装完整版本（包含额外依赖）
pip install gensim[complete]
'''

from Util.EmbeddingUtil import *
# 在加载模型后添加自定义词典
jieba.load_userdict("./Dict/custom_dict.txt")  # 文件中包含: 花呗

sentences = ['如何更换花呗绑定银行卡', '花呗更改绑定银行卡']

for sentence in sentences:
    x = text_to_embedding(sentence)