固有表現抽出(NER)を動かしてみる3。(windows, python)。

固有表現抽出(NER)を動かしてみる3。(windows, python)。

参考にしたサイト

参考にしたというより、そのまま、動かしただけ。

https://github.com/NielsRogge/Transformers-Tutorials

Custom_Named_Entity_Recognition_with_BERT.ipynb

https://github.com/NielsRogge/Transformers-Tutorials/blob/master/BERT/Custom_Named_Entity_Recognition_with_BERT.ipynb

結果

環境

windows10
python3.7

結果

時短のため、データを減らしたりしているので、全然、正しく動いてませんが。。。

              precision    recall  f1-score   support

         geo       0.44      0.50      0.47         8
         gpe       0.50      0.67      0.57         6
         org       0.33      0.33      0.33         3
         per       0.00      0.00      0.00         1
         tim       0.00      0.00      0.00         1

   micro avg       0.45      0.47      0.46        19
   macro avg       0.26      0.30      0.28        19
weighted avg       0.40      0.47      0.43        19

india has a capital called mumbai . on wednesday , the president will give a presentation
['B-gpe', 'O', 'O', 'O', 'O', 'B-geo', 'O', 'O', 'B-tim', 'O', 'O', 'O', 'O', 'O', 'O', 'O']

関連する自分の記事

ai-de-seikei.hatenablog.com

ai-de-seikei.hatenablog.com

コメント

何かアドバイスなどあれば、お願いします。