検索
連載

[文章生成]PyTorchのRNNクラスを使って文章を生成してみよう作って試そう! ディープラーニング工作室(2/2 ページ)

RNNを組み込んだニューラルネットワークを定義して、そこに分かち書きされたテキストから作成したデータを入力して学習を行い、文章がうまく生成されるかを見てみます。

PC用表示 関連情報
Share
Tweet
LINE
Hatena
前のページへ |       

1つの形態素から文章を生成してみる

 実際に文章を生成する前に、前回同様、その手順を確認しておきましょう。そのコードが以下です。ここではCPUとGPUのどちらを使うかをNetクラスのインスタンスから得た後に、それを評価モードに変更してから、「私」の次にどんな形態素がくるかを推測するようにしています。コードが長いので順番に見ていきましょう(今回のノートブックには以下のコードをまとめて1つのセルに記述してあります)。

前のページへ |       

Copyright© Digital Advantage Corp. All Rights Reserved.

ページトップに戻る