始める無料で始める

書籍タイトルのワンホットエンコード

PyBooks は、ライブラリ内の書籍ジャンルを整理・分析したいと考えています。書籍ジャンルのリストにワンホットエンコードを適用し、機械が扱える形式に変換しましょう。

torch はすでにインポートされています。

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

PyTorch で学ぶテキストの Deep Learning

コースを見る

演習の手順

  • 語彙のサイズを定義して、vocab_size に保存します。
  • 適切な torch の手法と vocab_size を使ってワンホットベクトルを作成します。
  • 辞書内包表記を使って、ジャンルをキー、そのワンホットベクトルを値とする対応表(辞書)を作成します。辞書のキーはジャンル名にしてください。

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

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

genres = ['Fiction','Non-fiction','Biography', 'Children','Mystery']

# Define the size of the vocabulary
vocab_size = ____(____)

# Create one-hot vectors
one_hot_vectors = torch.____(____)

# Create a dictionary mapping genres to their one-hot vectors
one_hot_dict = {____: ____[i] for i, genre in enumerate(genres)}

for genre, vector in one_hot_dict.items():
    print(f'{genre}: {vector.numpy()}')
コードを編集して実行