ONNX 模型库
返回模型

说明文档

Qdrant/Splade_PP_en_v1 的 ONNX 端口,用于文本分类和相似度搜索。

使用方法

以下是使用 FastEmbed 对模型进行推理的示例。

from fastembed import SparseTextEmbedding

documents = [
    "You should stay, study and sprint.",
    "History can only prepare us to be surprised yet again.",
]

model = SparseTextEmbedding(model_name="prithivida/Splade_PP_en_v1")
embeddings = list(model.embed(documents))

# [
#     SparseEmbedding(values=array(
#         [0.45940185, 0.64054322, 0.2425732, 0.1623179, 1.20566428,
#          0.62039357...]),
#                     indices=array([1012, 1998, 2000, 2005, 2017, 2022...])),
#     SparseEmbedding(values=array([
#         0.09767706, 0.4374367, 0.00468039, 1.01167965, 1.02318227, 1.30155718
#     ...]),
#                     indices=array([2017, 2022, 2025, 2057, 2064, 2069...]))
# ]

Qdrant/Splade_PP_en_v1

作者 Qdrant

sentence-similarity transformers
↓ 71.1K ♥ 6

创建时间: 2024-03-13 03:50:07+00:00

更新时间: 2024-07-15 12:55:11+00:00

在 Hugging Face 上查看

文件 (9)

.gitattributes
README.md
config.json
generation_config.json
model.onnx ONNX
special_tokens_map.json
tokenizer.json
tokenizer_config.json
vocab.txt