N3ER (Neural Network based Named Entity Recognizer) はBiLSTMとCRFを用いた固有表現認識システムです。 N3ERはBIO2 (Input, Output, Beginning) 形式を採用し,Pytorchで実装されています。
モデルの学習は以下のコマンドで実行します。
python train.py CONFIG
ここでCONFIGとは学習に必要な各種ディレクトリやモデルのハイパーパラメータを記述したファイルを指します。CONFIGのサンプルについてはこちらを参照ください。
また,モデルのテストは以下のコマンドで実行します。
python test.py CONFIG -m MODEL
ここでMODELは学習済みモデルの保存ディレクトリを指し,vierbiアルゴリズムを用いることでタグ推定を行います。基本的には推定したタグのみを出力しますが,引数 -d にデリミタを指定することで元の単語と併せて出力することが出来ます。
Last Modified: 2021-01-19 by Shirai