Comet T5 Base Japanese

🚀 COMET-T5 ja

COMET-T5 jaは、ATOMIC ja データセットを用いて、テキスト生成型の言語モデルとしてファインチューニングされたT5モデルです。このモデルは この論文 で紹介されています。

🚀 クイックスタート

このモデルは、テキスト生成パイプラインを使って直接利用することができます。生成結果にはランダム性が含まれるため、再現性を担保するためにシード値を設定します。

基本的な使用法

>>> from transformers import pipeline, set_seed
>>> generator = pipeline('text2text-generation', model='nlp-waseda/comet-t5-base-japanese')
>>> set_seed(42)
>>> generator("次の出来事の後に起こりうることは何ですか: Xが友人に電話する", max_length=30, num_return_sequences=5, do_sample=True)

[{'generated_text': 'Xが友人から返事を得る'},
 {'generated_text': 'Xが会話する'},
 {'generated_text': 'Xが友人に怒られる'},
 {'generated_text': 'Xが退屈しそうな雰囲気になる'},
 {'generated_text': 'Xが友人と会う'}]

📚 ドキュメント

前処理

各関係に対するプロンプトは異なります。

評価結果

このモデルは以下の評価結果を達成しています。

BibTeXエントリと引用情報

@InProceedings{ide_nlp2023_event,
    author =    "井手竜也 and 村田栄樹 and 堀尾海斗 and 河原大輔 and 山崎天 and 李聖哲 and 新里顕大 and 佐藤敏紀",
    title =     "人間と言語モデルに対するプロンプトを用いたゼロからのイベント常識知識グラフ構築",
    booktitle = "言語処理学会第29回年次大会",
    year =      "2023",
    url =       "https://www.anlp.jp/proceedings/annual_meeting/2023/pdf_dir/B2-5.pdf"
}