metadata
tags:
- bert
- token-classification
- sequence-tagger-model
language: en
widget:
- text: Total exports of maize
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)