metadata
language:
- ja
license: apache-2.0
library_name: transformers
datasets:
- llm-book/JGLUE
pipeline_tag: text-classification
bert-base-japanese-v3-jnli
「大規模言語モデル入門」の第5章で紹介している(自然言語推論)のモデルです。 cl-tohoku/bert-base-japanese-v3をJGLUEのMARC-jaデータセットでファインチューニングして構築されています。
関連リンク
使い方
from transformers import pipeline
nli_pipeline = pipeline(model="llm-book/bert-base-japanese-v3-jnli")
text = "二人の男性がジェット機を見ています"
entailment_text = "ジェット機を見ている人が二人います"
# textとentailment_textの論理関係を予測
print(nli_pipeline({"text": text, "text_pair": entailment_text}))
# {'label': 'entailment', 'score': 0.9964311122894287}