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: 106

Inference Providers NEW

Token Classification

This model is not currently available via any of the supported Inference Providers.