nljubesi commited on
Commit
4bd46a9
1 Parent(s): 3baf9f4

Added comment on the order of labels for ST

Browse files
Files changed (1) hide show
  1. README.md +9 -0
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
+ ```