Stsb TinyBERT L4

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

該模型用於預測兩個句子之間的語義相似度，藉助SentenceTransformers庫進行訓練，具有高效準確的特點。

🚀 快速開始

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

📦 安裝指南

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

💻 使用示例

基礎用法

from sentence_transformers import CrossEncoder

model = CrossEncoder('cross-encoder/stsb-TinyBERT-L4')
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 許可證。