gabrielandrade2
commited on
Commit
•
7316338
1
Parent(s):
a459477
Replace "cl-tohoku" with "tohoku-nlp"
Browse files- config.json +1 -1
- predict.py +1 -1
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"BertForTokenClassification"
|
5 |
],
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "tohoku-nlp/bert-base-japanese-whole-word-masking",
|
3 |
"architectures": [
|
4 |
"BertForTokenClassification"
|
5 |
],
|
predict.py
CHANGED
@@ -148,7 +148,7 @@ def run(model, input, output=None, normalize=False, **kwargs):
|
|
148 |
bert_tc = classification_model.bert_tc.to(device)
|
149 |
|
150 |
tokenizer = ner.NER_tokenizer_BIO.from_pretrained(
|
151 |
-
'
|
152 |
num_entity_type=len_num_entity_type # Entityの数を変え忘れないように!
|
153 |
)
|
154 |
|
|
|
148 |
bert_tc = classification_model.bert_tc.to(device)
|
149 |
|
150 |
tokenizer = ner.NER_tokenizer_BIO.from_pretrained(
|
151 |
+
'tohoku-nlp/bert-base-japanese-whole-word-masking',
|
152 |
num_entity_type=len_num_entity_type # Entityの数を変え忘れないように!
|
153 |
)
|
154 |
|