BLEURT 20 D12

🚀 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-D12')
model = BleurtForSequenceClassification.from_pretrained('lucadiliello/BLEURT-20-D12')
tokenizer = BleurtTokenizer.from_pretrained('lucadiliello/BLEURT-20-D12')

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.9604414105415344, 0.8080050349235535]

若想了解 BleurtConfig、BleurtForSequenceClassification 和 BleurtTokenizer 在PyTorch中的定義，請查看 此倉庫。