Downloads · 30 days
14
3% of all-time downloads
autotrain_gro_ner
Wanjiru/autotrain_gro_ner
autotrain_gro_ner is a token classification model from Wanjiru. Use it when you need labels on individual words, such as names. It is set up for transformers.
All-time downloads
509
Public
Repo size
2.7 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.bin1.3 GB · 100%
At a glance
- Task
- Token Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Jul 18, 2022
- Updated
- Sep 21, 2022
- SHA
- ff23b09d
Try a prompt
Model card
From the Hugging Face model README
Token Classification
Classifies Gro's items and metrics
| tag | token |
|---|---|
| B-ITEM | BEGINNING ITEM |
| I-ITEM | INSIDE ITEM |
| B-METRIC | BEGINNING METRIC |
| I-METRIC | INSIDE METRIC |
| O | OUTSIDE |
Training: Script to train this model
The following Flair script was used to train this model:
from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
tokenizer = AutoTokenizer.from_pretrained("Wanjiru/autotrain_gro_ner")
model = AutoModelForTokenClassification.from_pretrained("Wanjiru/autotrain_gro_ner")
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "Wanjru"
ner_res = nlp(example)
- Task
- Token Classification
- Library
- transformers
- Type
- bert
- Languages
- en
- Created
- Jul 18, 2022
- Updated
- Sep 21, 2022