How do we customize the number of entites for NER pretrained model?
Hi,I am trying to re-train the model with custom dataset with X entites. I found the documentation link:AutoModels — transformers 3.0.2 documentationAutoModelForTokenClassification.from_pretrained(“bert-base-multilingual-cased”)Anyone knows what is the configuration to set the pre-trained huggingface model to X entities?Thank you,Dinesh
2022-10-06T18:59:40Z
AutoModelForTokenClassification.from_pretrained(“bert-base-multilingual-cased”, num_labels=X)