这是一个预训练模型,用于处理印地语-英语混合代码数据的名称实体识别,来自 LinCE
该模型是根据下面的存储库进行训练的。
https://github.com/sagorbrur/codeswitch
要安装codeswitch,请执行以下操作:
pip install codeswitch
from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline tokenizer = AutoTokenizer.from_pretrained("sagorsarker/codeswitch-hineng-ner-lince") model = AutoModelForTokenClassification.from_pretrained("sagorsarker/codeswitch-hineng-ner-lince") ner_model = pipeline('ner', model=model, tokenizer=tokenizer) ner_model("put any hindi english code-mixed sentence")
from codeswitch.codeswitch import NER ner = NER('hin-eng') text = "" # your mixed sentence result = ner.tag(text) print(result)