alfred-ner / README.md
sammmmmmm25's picture
Upload README.md with huggingface_hub
1d46def verified
|
Raw
History Blame Contribute Delete
619 Bytes
---
language: en
tags: [ner, token-classification, criminal-justice, alfred-ai]
license: mit
base_model: dslim/bert-base-NER
---
# Alfred NER
Fine-tuned from `dslim/bert-base-NER` for criminal justice entity extraction.
## Labels
PER 路 ORG 路 LOC 路 DATE 路 VEH 路 WEAPON 路 CASE 路 ALIAS
## Usage
```python
from transformers import pipeline
ner = pipeline("ner", model="sammmmmmm25/alfred-ner", aggregation_strategy="simple")
ner("Suspect Kevin Truro fled in a white Honda Accord on Route 9.")
```
## Training
- Base model: `dslim/bert-base-NER`
- Epochs: 10 | Batch size: 16 | fp16: True
- Validation F1: 0.8646