metadata
language:
- en
thumbnail: url to a thumbnail used in social sharing
tags:
- tag1
- tag2
license: cc
datasets:
- conll2003
Model information:
distilibert-base-uncased model finetuned using the conll2003 dataset from the datasets library.
Intended uses & limitations
This model is intended to be used for named entity recoginition tasks. The model will identify entities of persons, locations, organisations, and miscellaneous. The model will predict lables based upon the CoNLL-2003 dataset that has the following class labels - ['O', 'B-PER', 'I-PER', 'B-ORG', 'I-ORG', 'B-LOC', 'I-LOC', 'B-MISC', 'I-MISC']
Note that the dataset and model may not be fully represetative or suitable for all needs it is recommended that the paper for the dataset and base model card should be reviewed before using the model -
How to use
Load the model from the library
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("sarahmiller137/distilbert-base-uncased-ft-conll2003")
model = AutoModel.from_pretrained("sarahmiller137/distilbert-base-uncased-ft-conll2003")