learnrr commited on
Commit
4df126f
·
verified ·
1 Parent(s): 5e2d71d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -1
README.md CHANGED
@@ -25,4 +25,31 @@ model-index:
25
  metrics:
26
  - type: f1
27
  value: 95.99
28
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
  metrics:
26
  - type: f1
27
  value: 95.99
28
+ ---
29
+
30
+ ## Model description
31
+ This model is a fine-tuned version of roberta-base for the Named Entity Recognition (NER) task using the CoNLL-2003 dataset. It can identify four types of entities: Persons (PER), Organizations (ORG), Locations (LOC), and Miscellaneous (MISC).
32
+ ## Training procedure
33
+ * **Hardware:** NVIDIA V100 GPU
34
+ * **Optimizer:** AdamW
35
+ * **Learning Rate:** 2e-5
36
+ * **Batch Size:** 16
37
+ * **Weight Decay:** 0.01
38
+ * **Epochs:** 5
39
+ * **Mixed Precision Training:** FP16 enabled
40
+ ## Evaluation Results
41
+ | Metric) | Value |
42
+ | :--- | :--- |
43
+ | **F1 Score** | **95.99%** |
44
+ | **Precision** | **95.61%** |
45
+ | **Recall** | **96.38%** |
46
+ | **Accuracy** | **99.29%** |
47
+ | **Eval Loss** | **0.0464** |
48
+ ## How to use
49
+ ```python
50
+ from transformers import pipeline
51
+ model_id = "learnrr/roberta-NER-conll2003"
52
+ text = "Apple is looking at buying U.K. startup for $1 billion"
53
+ results = nlp(text)
54
+ for entity in results:
55
+ print(f"entity: {entity['word']} | class: {entity['entity_group']} | confidence: {entity['score']:.4f}")