metadata
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
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