問題タブ [bert-language-model]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票する
1 に答える
1964 参照

named-entity-recognition - トレーニング済みの BERT NER (固有表現認識) モデルを使用して新しい例を予測するにはどうすればよいですか?

このミディアムポストに続いて、私は自分のBERT NERを訓練しました:https://medium.com/@yingbiao/ner-with-bert-in-action-936ff275bc73

モデルをディスクに保存し、正常にロードしました。

model.eval() は動作します:

私はBERTとTransformer libが初めてです。私は似たようなものを期待します

認識されたエンティティが表示されます。

私も試しました:

出力が私に大きなテンソルを与え、それをどうするかわかりません。

このモデルを使用して、例文のエンティティを予測するにはどうすればよいですか? アウトプットはどうすればいいの?

ありがとう!