Description
A Named Entity Recognition model trained on a customer feedback data using DistilBert. Possible labels are:
- PRD: for certain products
- BRND: for brands
Usage
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_ner_v6_distil")
model = AutoModelForTokenClassification.from_pretrained("CouchCat/ma_ner_v6_distil")
- Downloads last month
- 5