始める無料で始める

AutoTokenizer によるテキストの前処理

農家が現場で直面する問題について質問できる、精密農業アプリケーションを構築しています。農家が実際に抱えるよくある質問と回答のデータセットを活用します。このデータセットには以下のフィールドが含まれています。

  • question:農業に関するよくある質問
  • answers:農業に関する質問への回答

分散学習の第一ステップとして、このテキストデータセットの前処理を行いましょう。

以下のデータはあらかじめ読み込まれています。

  • dataset:農業に関する質問と回答のサンプルデータセット
  • AutoTokenizertransformers からインポート済み

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

PyTorch による効率的な AI モデルトレーニング

コースを見る

演習の手順

  • 事前学習済みの tokenizer を読み込みましょう。
  • tokenizer を使って example["question"] をトークン化しましょう。
  • encode() 関数を dataset に適用しましょう。

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

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

# Load a pre-trained tokenizer
tokenizer = ____.____("distilbert-base-uncased")

def encode(example):
    # Tokenize the "question" field of the training example
    return ____(____["____"], padding="max_length", truncation=True, return_tensors="pt")

# Map the function to the dataset
dataset = ____.____(____, batched=True)

dataset = dataset.map(lambda example: {"labels": example["answers"]}, batched=True)

print(dataset)
コードを編集して実行