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

Connected

演習

感情分析CNNモデルのテスト

モデルの学習が完了したので、PyBooks は新しい書評で性能を確認したいと考えています。

レビューの感情がポジティブかネガティブかを判定します。

次のパッケージはインポート済みです: torch、torch.nn は nn、torch.nn.functional は F、torch.optim は optim。

vocab_size と embed_dim を引数にした TextClassificationCNN() のインスタンスも読み込まれており、model として保存されています。

指示

100 XP
  • book_reviews リストを反復し、各レビュー内の単語をテンソルに変換します。
  • 各 input_tensor に対するモデルの出力を取得します。
  • outputs.data から、最も可能性の高い感情カテゴリのインデックスを見つけます。
  • predicted_label の要素を取り出して変換し、1 を "Positive" ラベルとする感情文字列にします。