今回は、Transformersのインストールとテキスト分類を行います。
Transformersのインストール
Transformersのインストールを行うためには次のコマンドを実行します。
実行環境としてはGoogle Colaboratoryを想定しています。
[Google Colaboratory]
1 | !pip install transformers[ja]==4.4.2 |
次のような実行結果が表示されればインストールは成功しています。
[実行結果]
1 | Collecting transformers[ja]==4.4.2 |
テキスト分類
テキスト分類は、テキストを事前に定義されたカテゴリに分類する処理となります。
今回は文章を「ポジティブ」か「ネガティブ」かに分類してみます。
タスク名に‘sentiment-analysis’を指定したパイプラインを作成し、そのパイプラインに任意の文章を渡します。
[Google Colaboratory]
1 | from transformers import pipeline |
[実行結果]
1 | Downloading: 100% |
スコア99以上で‘POSITIVE’な文章であると判断できました。
少し文章を変更して再度テキスト分類を行います。
[Google Colaboratory]
1 | from transformers import pipeline |
[実行結果]
1 | [{'label': 'NEGATIVE', 'score': 0.9983341693878174}] |
スコア99以上で‘NEGATIVE’な文章であると判断できました。
次回は、質問応答を行います。