库名称: light-embed
流水线标签: 句子相似度
标签:

• 句子转换器
• 特征提取
• 句子相似度

onnx-models/all-MiniLM-L6-v2-onnx

这是sentence-transformers/all-MiniLM-L6-v2的ONNX移植版本，用于生成文本嵌入。

模型详情

• 嵌入维度: 384
• 最大序列长度: 256
• 磁盘文件大小: 0.08 GB
• ONNX中集成的模块: Transformer、池化、归一化

使用方法

安装light-embed后，使用此模型变得非常简单：

pip install -U light-embed

然后，您可以通过指定原始模型名称来使用模型，如下所示：

from light_embed import TextEmbedding  
sentences = [  
    "这是一个示例句子",  
    "每个句子都会被转换"  
]  

model = TextEmbedding('sentence-transformers/all-MiniLM-L6-v2')  
embeddings = model.encode(sentences)  
print(embeddings)  

或者通过指定onnx模型名称来使用模型，如下所示：

from light_embed import TextEmbedding  
sentences = [  
    "这是一个示例句子",  
    "每个句子都会被转换"  
]  

model = TextEmbedding('onnx-models/all-MiniLM-L6-v2-onnx')  
embeddings = model.encode(sentences)  
print(embeddings)  

引用与作者

Binh Nguyen / binhcode25@gmail.com