Stsb Distilroberta Base

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

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

🚀 快速開始

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

✨ 主要特性

• 基於預訓練模型，能夠預測兩個句子之間語義相似度的得分，得分範圍在 0 到 1 之間。

📦 安裝指南

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

💻 使用示例

基礎用法

from sentence_transformers import CrossEncoder

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

上述代碼展示瞭如何使用預訓練模型預測句子對的語義相似度得分。模型將為 ('Sentence 1', 'Sentence 2') 和 ('Sentence 3', 'Sentence 4') 這兩對句子預測得分。

高級用法

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

📚 詳細文檔

訓練數據

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

📄 許可證

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

📋 模型信息