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

演習

テキスト生成用の RNN モデルを作成する

PyBooks では、テキスト生成を行うアルゴリズムの開発を任されています。プロジェクトの目的は、本のタイトルのオートコンプリートです。まずはこのプロジェクトの立ち上げとして、Recurrent Neural Network (RNN) を試してみることにします。こうすることで、より複雑なモデルに進む前に RNN の特性を理解できます。

次のモジュールはすでにインポートされています：torch、torch.nn は nn として。

変数 data には、Lewis Carroll の Alice's Adventures in Wonderland（不思議の国のアリス）の抜粋が設定されています。

指示

100 XP
  • RNNmodel クラスに RNN レイヤーと線形レイヤーを含めます。
  • 入力サイズを chars の長さ、隠れ状態サイズを 16、出力サイズを chars の長さとして、RNN モデルをインスタンス化します。