Views
No views yet
[UNK] (unknown) エンティティを扱えるようにしました1from transformers import AutoTokenizer, AutoModel
2
3# 本モデル用のトークナイザのコードを使用するため、trust_remote_code=True の指定が必要です
4tokenizer = AutoTokenizer.from_pretrained("uzabase/luke-japanese-wordpiece-base", trust_remote_code=True)
5
6model = AutoModel.from_pretrained("uzabase/luke-japanese-wordpiece-base")position_ids を含めるように変更しました。
position_ids が使われていましたが、これは RoBERTa 仕様のものであり、BERT を使った本モデルでは正しい値となっていませんでした。そこで、 BERT 向けの正しい position_ids の値がモデルに入力されるように、position_ids を明示的にトークナイザの出力に含めるようにしました。entity_vocab の各トークン("[PAD]" 等の特殊トークンを除く)の先頭に付いていた "None:" の文字列を除去しました。
"None:聖徳太子" となっていたトークンは "聖徳太子" に修正されています。