始める無料で始める

事前学習済みGPT-2モデルでのテキスト補完

PyBooksに戻りましょう。現在のプロジェクトでは、既存の物語をもとに魅力的なストーリーを生成し、顧客を惹きつけて体験を高めることが目標です。そのためには、自然に続きが書ける強力なテキスト生成ツールが必要です。ここでは事前学習済みモデルを使って実現します。

次のライブラリは読み込まれています:torch と、transformers からの GPT2TokenizerGPT2LMHeadModel

この演習はコースの一部です

PyTorch で学ぶテキストの Deep Learning

コースを見る

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Initialize the tokenizer
tokenizer = ____

# Initialize the pre-trained model
model = ____
コードを編集して実行