Stsb Roberta Base

🚀 用於語義文本相似度的交叉編碼器

本模型利用 SentenceTransformers 的 Cross-Encoder 類進行訓練，可有效預測兩個句子之間的語義相似度。

🚀 快速開始

本模型使用 SentenceTransformers 的 Cross-Encoder 類進行訓練。

✨ 主要特性

• 使用特定的訓練數據，在語義文本相似度預測方面表現出色。
• 可以預測兩個句子之間語義相似度的得分，範圍在 0 到 1 之間。
• 既可以結合 sentence_transformers 使用，也可以使用 Transformers 的 AutoModel 類獨立使用。

📦 安裝指南

文檔未提及安裝步驟，跳過此章節。

💻 使用示例

基礎用法

from sentence_transformers import CrossEncoder

model = CrossEncoder('cross-encoder/stsb-roberta-base')
scores = model.predict([('Sentence 1', 'Sentence 2'), ('Sentence 3', 'Sentence 4')])

上述代碼展示瞭如何使用預訓練模型預測兩對句子的語義相似度得分。

高級用法

你可以不使用 sentence_transformers，僅通過 Transformers 的 AutoModel 類來使用此模型。不過文檔未給出具體代碼示例。

📚 詳細文檔

訓練數據

該模型在 STS 基準數據集 上進行訓練，能夠預測兩個句子語義相似度的得分，範圍在 0 到 1 之間。

模型信息

📄 許可證

本項目採用 Apache-2.0 許可證。