BLEURT 20

🚀 BLEURT-PyTorch模型

本模型基於自定義的Transformer模型，可用於特定的序列分類任務，能有效評估文本之間的相似度，在自然語言處理領域有廣泛應用價值。

🚀 快速開始

📦 安裝指南

你可以使用以下命令安裝該模型：

pip install git+https://github.com/lucadiliello/bleurt-pytorch.git

💻 使用示例

基礎用法

以下代碼展示瞭如何加載模型並進行預測：

import torch
from bleurt_pytorch import BleurtConfig, BleurtForSequenceClassification, BleurtTokenizer

config = BleurtConfig.from_pretrained('lucadiliello/BLEURT-20')
model = BleurtForSequenceClassification.from_pretrained('lucadiliello/BLEURT-20')
tokenizer = BleurtTokenizer.from_pretrained('lucadiliello/BLEURT-20')

references = ["a bird chirps by the window", "this is a random sentence"]
candidates = ["a bird chirps by the window", "this looks like a random sentence"]

model.eval()
with torch.no_grad():
    inputs = tokenizer(references, candidates, padding='longest', return_tensors='pt')
    res = model(**inputs).logits.flatten().tolist()
print(res)
# [0.9990496635437012, 0.7930182218551636]

📚 詳細文檔

有關BleurtConfig、BleurtForSequenceClassification和BleurtTokenizer在PyTorch中的定義，請查看此倉庫。