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

演習

RNN を使ったテキスト生成 - 学習と生成

PyBooks のチームは、与えられた入力から次の文字を予測して書名のオートコンプリートを行う RNN モデルを学習・テストしてほしいと考えています。このプロジェクトは、テキスト補完モデルのさらなる開発に役立ちます。

RNNmodel クラスの model インスタンスはあらかじめ読み込まれています。data 変数は前処理され、シーケンスとしてエンコードされています。

inputs と targets 変数もあらかじめ用意されています。

指示

100 XP
  • 本モデルの誤差を計算するために使用する損失関数をインスタンス化します。
  • PyTorch の最適化モジュールからオプティマイザをインスタンス化します。
  • モデルを train モードに設定し、最適化ステップの前に勾配をゼロにして、学習プロセスを実行します。
  • 学習後、サンプル入力でテストできるように、モデルを評価モードに切り替えます。