tejasvichebrolu commited on
Commit
0502d84
·
verified ·
1 Parent(s): 3b22585

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -0
README.md CHANGED
@@ -72,3 +72,31 @@ results = classifier(headlines)
72
  for result in results:
73
  print(f"Label: {result['label']}, Score: {result['score']:.4f}")
74
  ```
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
72
  for result in results:
73
  print(f"Label: {result['label']}, Score: {result['score']:.4f}")
74
  ```
75
+
76
+ ## Training Procedure
77
+
78
+ The model was fine-tuned on a dataset of **860 headlines** annotated for the presence of a *conflict frame*.
79
+
80
+ ---
81
+
82
+ ## Hyperparameters
83
+
84
+ | Hyperparameter | Value |
85
+ |---------------------------|---------------|
86
+ | Conflict Class Weight | 1.69 |
87
+ | Non-Conflict Class Weight | 9.01 |
88
+ | Learning Rate | 6.008 × 10⁻⁵ |
89
+ | Epochs | 9 |
90
+
91
+ ---
92
+
93
+ ## Evaluation Results
94
+
95
+ The model's performance was evaluated using **5-fold cross-validation**. The average metrics are reported below:
96
+
97
+ | Class | Precision | Recall | F1-Score |
98
+ |---------------|-----------|--------|----------|
99
+ | Conflict | 0.92 | 0.91 | 0.92 |
100
+ | Non-Conflict | 0.81 | 0.82 | 0.81 |
101
+ | Macro avg | 0.87 | 0.87 | **0.87** |
102
+