bert-base-japanese-v3-jnli

「大規模言語モデル入門」の第5章で紹介している(自然言語推論)のモデルです。 cl-tohoku/bert-base-japanese-v3をJGLUEのMARC-jaデータセットでファインチューニングして構築されています。

関連リンク

• GitHubリポジトリ
• Colabノートブック（訓練）
• Colabノートブック（推論）
• データセット
• 大規模言語モデル入門（Amazon.co.jp）
• 大規模言語モデル入門（gihyo.jp）

使い方

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}

ライセンス

Apache License 2.0