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

Connected

Exercise

テキスト用のCNNモデルを学習する

TextClassificationCNN クラスの定義はお見事です。次は PyBooks でモデルを学習し、書評の感情分析を高精度に行えるよう最適化していきます。

次のパッケージはすでにインポートされています: torch、torch.nn は nn、torch.nn.functional は F、torch.optim は optim。

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

Instructions

100 XP
  • 二値分類に用いる損失関数を定義し、criterion として保存します。
  • 学習ループの開始時に勾配をゼロにリセットします。
  • ループの最後でパラメータを更新します。