Added comment on the order of labels for ST
Browse files
README.md
CHANGED
|
@@ -41,3 +41,12 @@ If you use this fine-tuned model, please cite the following paper:
|
|
| 41 |
pages = "37--42",
|
| 42 |
}
|
| 43 |
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 41 |
pages = "37--42",
|
| 42 |
}
|
| 43 |
```
|
| 44 |
+
|
| 45 |
+
When running the model in `simpletransformers`, the order of labels has to be set as well.
|
| 46 |
+
|
| 47 |
+
```
|
| 48 |
+
from simpletransformers.ner import NERModel, NERArgs
|
| 49 |
+
model_args = NERArgs()
|
| 50 |
+
model_args.labels_list = ['B-LOC','B-MISC','B-ORG','B-PER','I-LOC','I-MISC','I-ORG','I-PER','O']
|
| 51 |
+
model = NERModel('electra', 'classla/bcms-bertic-ner', args=model_args)
|
| 52 |
+
```
|