学習したモデルを使って要約を行います。
要約
要約を行うコードは以下の通りです。
5行目で、日本語T5事前学習済みモデルのトークナイザーを読み込んでいます。
6行目では、前回ファインチューニングした要約モデルをoutputフォルダから読み込んでいます。
9行目に、要約対象の文章を設定しています。
[Google Colaboratory]
1 | import torch |
要約結果は以下の通りです。
[実行結果]
1 | /usr/local/lib/python3.7/dist-packages/torch/_tensor.py:575: UserWarning: floor_divide is deprecated, and will be removed in a future version of pytorch. It currently rounds toward 0 (like the 'trunc' function NOT 'floor'). This results in incorrect rounding for negative values. |
・・・要約と言えば要約されていますが、主語が全部とんでますし、中盤以降を全部省略とずいぶん大胆な要約となっています。
なんらかの改善が必要なのかもしれません。