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中的定义，请查看 此仓库。