heyong.fu
22 lines
826 B
Python
22 lines
826 B
Python
# -*- coding: utf-8 -*-
|
|
# 说明：验证 Sentence Transformers 是否安装成功
|
|
|
|
# 说明：导入 SentenceTransformer 类
|
|
from sentence_transformers import SentenceTransformer
|
|
|
|
# 说明：尝试加载一个轻量级模型进行测试
|
|
# "all-MiniLM-L6-v2" 是一个小型的通用模型，适合快速测试
|
|
# 首次运行时会自动下载模型（可能需要一些时间）
|
|
print("正在加载模型进行测试...")
|
|
model = SentenceTransformer("all-MiniLM-L6-v2")
|
|
|
|
# 说明：对一个简单的句子进行编码测试
|
|
sentence = "这是一个测试句子"
|
|
embedding = model.encode(sentence)
|
|
|
|
# 说明：检查嵌入向量的形状
|
|
print(f"安装成功！嵌入向量维度：{embedding.shape}")
|
|
print(f"前 5 个维度值：{embedding[:5]}")
|
|
|
|
# 说明：如果没有报错并输出了维度信息，说明安装成功
|