CouchCat
/

ma_ner_v6_distil

Token Classification Transformers PyTorch English distilbert ner Inference Endpoints

Model card Files Files and versions Community

Edit model card

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