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")
New

Select AutoNLP in the “Train” menu to fine-tune this model automatically.

Downloads last month
10
Hosted inference API
Token Classification
This model can be loaded on the Inference API on-demand.