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での定義については、このリポジトリ を参照してください。