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
40
Hosted inference API
Token Classification