sarahmiller137's picture
Add model card
16ac3ab
---
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 -
- [Conll2003](https://aclanthology.org/W03-0419)
- [distilbert](https://huggingface.co/distilbert-base-uncased)
## How to use
Load the model from the library
```python
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")
```