Shingoの数学ノート

プログラミングと機械学習のメモ

ポジネガ可視化アプリ「BERTSHAP」を作成しました！

日付：    カテゴリ： 自然言語処理

今回は、ポジネガ可視化アプリである「BERTSHAP」を作成しましたので、紹介します。

BERTSHAPとは？

仕組みはBERTとshapでポジネガ分析の結果を解釈するで解説しています。

なお、今回のモデルはモデルの軽量化+予測の高速化を実現したかったため、BERTの軽量版のDistilBERTを使用しています。

テキストボックスに「今日のランチは最高だった！」と入力してみます。

すると、以下のように表示されます。

「最高」の部分が赤くなっていて、プラスに働いていることが一目でわかります。

実行の動画の様子は以下をみてください。

BERTSHAPのインストール

今回は、docker版のみの提供です。

基本はcpu版をインストールしてください。もし、gpuが搭載されているpcで動かすのであれば、gpuもおすすめです。

cpu版

docker run -it -p 8000:8000 shingo425/bertshap-cpu

gpu版

docker run -it -p 8000:8000 shingo425/bertshap-gpu

あとは、ブラウザのURLに「localhost:8000」を入力するとアプリが起動します。

感想など

自然言語処理の予測で、ただ予測するだけでなく、どの部分が効いているかを示すアプリを作りたかったので、 完成してとてもよかったです。

ぜひ使ってみていただけると嬉しいです。


Comment Box is loading comments...