正しいモデル構造を使用する
質問応答は抽出型または生成型のいずれかがあり、それぞれ入力と出力を正しく処理するために異なるトランスフォーマー構造が必要です。
それらは次のいずれかを使用します:
- エンコーダー専用モデルのような
"distilbert-base-uncased-distilled-squad" - デコーダー専用モデルのような
"gpt2"
特定のタスクでよく使われるモデルの知識を活かして、適切なものを選びましょう。パイプラインが読み込まれ、モナ・リザに関するテキストも読み込まれます。
この演習はコースの一部です
Python で学ぶ LLM の入門
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
question = "Who painted the Mona Lisa?"
# Define the appropriate model
qa = pipeline(task="question-answering", model=____)
output = qa(question=question, context=text)
print(output['answer'])