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

» 2021年03月26日 05時00分 公開
[かわさきしんじDeep Insider編集部]
前のページへ 1|2       

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

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

前のページへ 1|2       

Copyright© Digital Advantage Corp. All Rights Reserved.

RSSについて

アイティメディアIDについて

メールマガジン登録

@ITのメールマガジンは、 もちろん、すべて無料です。ぜひメールマガジンをご購読ください。