1. 学ぶ
    2. /
  2. コース
    3. /
  3. PyTorch で学ぶテキストの Deep Learning
Connected

演習

PyTorchでの埋め込み

PyBooks では書籍レコメンドシステムがうまく機能していましたが、テキストに含まれる一部のセマンティクスを考慮できていません。PyTorch の組み込み埋め込みレイヤーは、データから直接、単語同士の関係を学習・表現できます。チームはこの機能を使ってレコメンドを改善できるか検討中です。実装を手伝っていただけますか？

torch と、torch.nn は nn としてインポート済みです。

指示

100 XP
  • words の各単語に一意のインデックスを割り当て、word_to_idx に保存します。
  • word_to_idx を PyTorch のテンソルに変換し、inputs に保存します。
  • torch モジュールを使って、次元数が 10 の埋め込みレイヤーを初期化します。
  • 埋め込みレイヤーに inputs テンソルを渡し、出力を確認します。