Update README.md
Browse files
README.md
CHANGED
@@ -10,4 +10,48 @@ base_model:
|
|
10 |
tags:
|
11 |
- event_extraction
|
12 |
- email
|
13 |
-
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
tags:
|
11 |
- event_extraction
|
12 |
- email
|
13 |
+
---
|
14 |
+
# DistilBERT-based Named Entity Recognition (NER) Model for Event Extraction
|
15 |
+
|
16 |
+
This repository contains a fine-tuned `DistilBERT` model for extracting **event-related entities** such as:
|
17 |
+
|
18 |
+
- **Event Name**
|
19 |
+
- **Date**
|
20 |
+
- **Time**
|
21 |
+
- **Venue**
|
22 |
+
|
23 |
+
The model is trained on a custom NER dataset with the goal of accurately parsing event details from plain text, such as emails or notifications.
|
24 |
+
|
25 |
+
---
|
26 |
+
|
27 |
+
## Model Details
|
28 |
+
|
29 |
+
- **Base Model:** `distilbert-base-uncased`
|
30 |
+
- **Task:** Token Classification (NER)
|
31 |
+
- **Fine-Tuned On:** Custom annotated dataset for events
|
32 |
+
- **Framework:** [HuggingFace Transformers](https://huggingface.co/transformers/)
|
33 |
+
- **Library Versions:**
|
34 |
+
- `transformers`: 4.25+
|
35 |
+
- `datasets`: 2.10+
|
36 |
+
- `torch`: 1.13+
|
37 |
+
- `seqeval`: for evaluation metrics
|
38 |
+
|
39 |
+
---
|
40 |
+
|
41 |
+
## Labels Used
|
42 |
+
|
43 |
+
```text
|
44 |
+
EVENT_NAME
|
45 |
+
DATE
|
46 |
+
TIME
|
47 |
+
VENUE
|
48 |
+
O # Outside any entity
|
49 |
+
```
|
50 |
+
---
|
51 |
+
|
52 |
+
## Contributors
|
53 |
+
|
54 |
+
- [Thiyaga158](https://huggingface.co/Thiyaga158)
|
55 |
+
- [Hansika08](https://huggingface.co/Hansika08)
|
56 |
+
|
57 |
+
---
|