Downloads · 30 days
12
0% of all-time downloads
ma_ner_v7_distil
CouchCat/ma_ner_v7_distil
ma_ner_v7_distil is a token classification model from CouchCat. Use it when you need labels on individual words, such as names. It is set up for transformers. The card lists the license as mit.
A Named Entity Recognition model trained on a customer feedback data using DistilBert. Possible labels are in BIO-notation. Performance of the PERS tag could be better because of low data samples:
All-time downloads
2.7K
Public
Repo size
1 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin261 MB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- License
- mit
- Model type
- distilbert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Feb 28, 2021
- SHA
- 9dd0c9b1
Try a prompt
Model card
From the Hugging Face model README
Description
A Named Entity Recognition model trained on a customer feedback data using DistilBert. Possible labels are in BIO-notation. Performance of the PERS tag could be better because of low data samples:
- PROD: for certain products
- BRND: for brands
- PERS: people names
The following tags are simply in place to help better categorize the previous tags
- MATR: relating to materials, e.g. cloth, leather, seam, etc.
- TIME: time related entities
- MISC: any other entity that might skew the results
Usage
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("CouchCat/ma_ner_v7_distil")
model = AutoModelForTokenClassification.from_pretrained("CouchCat/ma_ner_v7_distil")
- Task
- Token Classification
- Library
- transformers
- Type
- distilbert
- License
- mit
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Feb 28, 2021