ner_distillbert_v1

ner-distilbert-base-cased

This is a Named Entity Recognition (NER) model based on the distilbert-base-cased architecture. This model was expertly fine-tuned to serve a critical function: accurately identifying and classifying key information within the specific domain of child helpline conversations. Through its specialized training, the model has been optimized to excel at recognizing the following entities crucial for this context:

`CALLER`: The person initiating the call for help.
`COUNSELOR`: The helpline staff member providing assistance.
`VICTIM`: The person who is the subject of the issue being discussed.
`PERPETRATOR`: The person causing the harm or issue.
`LOCATION`: Geographic places relevant to the conversation.
`AGE`: The age of the individuals involved.
`GENDER`: Gender references.
`INCIDENT_TYPE`: The specific type of problem or issue being reported (e.g., "bullying", "abuse"
   ).

Model Details

• Model Name: ner-distilbert-base-cased
• Version: 1
• Task: Ner
• Languages: en
• Framework: sklearn
• License: apache-2.0

Intended Uses & Limitations

Scope of Use

In Scope: The model is intended for text that is structurally and contextually similar to the training data (i.e., conversational, first-person accounts of issues). Out of Scope: This model is not intended for general-purpose NER tasks (e.g., analyzing news articles, legal documents, or emails). Its performance on text outside of its specialized domain has not been evaluated.

Training Data

• Dataset: ner_dataset_1.jsonl
• Size: Not specified
• Languages: en
• Nature: The training data is synthetic, generated to mimic real-world conversations while protecting the privacy and confidentiality of actual helpline users.

Training Configuration

Performance Metrics

Evaluation Results

Usage

Installation

pip install transformers torch

Named Entity Recognition Example

from transformers import pipeline

Use the model from the openchs repository on Hugging Face
  ner = pipeline("ner", model="openchs/ner_distillbert_v1", aggregation_strategy="simple")

  Example of a typical helpline conversation snippet
  text = "Hello 116, my name is Mary and I'm calling from Kampala. My daughter, Jane, is 12 years old and is being
  bullied at school by a boy named Peter."

  entities = ner(text)
  for entity in entities:
      print(f"{entity['entity_group']}: {entity['word']} (score: {entity['score']:.2f})")

MLflow Tracking

• Experiment: NER_Distilbert/marlon
• Run ID: 10d2648a456a4f6ab74022a9e45c9f40
• Training Date: 2025-10-30 11:58:48
• Tracking URI: http://192.168.10.6:5000

Training Metrics Visualization

View detailed training metrics and TensorBoard logs in the Training metrics tab.

Citation

@misc{ner_distilbert_base_cased,
  title={ner-distilbert-base-cased},
  author={OpenCHS Team},
  year={2025},
  publisher={Hugging Face},
  url={https://huggingface.co/openchs/ner_distillbert_v1}
}

Contact

[email protected]

---

Model card auto-generated from MLflow