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

演習

BERTモデルを評価する

BERT のトークナイザでサンプルレビューをトークン化したので、次は PyBooks のサンプルで BERT モデルを評価します。さらに、新しいデータに対するモデルの感情予測も評価します。

次のものはインポート済みです：BertTokenizer、BertForSequenceClassification、torch。 学習済みの model インスタンスも読み込まれています。これを新しいデータサンプルでテストします。

指示

100 XP
  • 評価用テキストをトークン化し、PyTorch テンソルを返すようにしてモデルに渡します。
  • 出力の logits を 0〜1 の確率に変換します。
  • その確率から感情を表示します。