star092304 commited on
Commit
b8d5efd
·
verified ·
1 Parent(s): 44c73ae

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -1
README.md CHANGED
@@ -12,4 +12,33 @@ metrics:
12
  - qwk
13
  - rmse
14
  pipeline_tag: text-classification
15
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  - qwk
13
  - rmse
14
  pipeline_tag: text-classification
15
+ ---
16
+
17
+ # IELTS Writing Task 2 Essay Scorer (DeBERTa-v3-base)
18
+
19
+ This repository contains a custom **Prompt-Aware IELTS Essay Scorer** fine-tuned on DeBERTa-v3-base. It simultaneously predicts the 4 analytical criteria of IELTS Writing Task 2 criteria and computes the overall band score.
20
+
21
+ ## 🔗 Quick Links & Resources
22
+ * **Dataset:** [View Clean Dataset](https://huggingface.co/star092304/ielts-writing-task2-debertav3base/tree/main/data)
23
+ * **Training Logs:** [View training_history.csv](https://huggingface.co/star092304/ielts-writing-task2-debertav3base/blob/main/training_history.csv)
24
+ * **Training Notebook:** [src/deberta-v3-base-ielts-aes.ipynb](https://huggingface.co/star092304/ielts-writing-task2-debertav3base/blob/main/src/deberta-v3-base-ielts-aes.ipynb)
25
+ * **Inference Notebook:** [src/IELTS_DeBERTa_Inference.ipynb](https://huggingface.co/star092304/ielts-writing-task2-debertav3base/blob/main/src/IELTS_DeBERTa_Inference.ipynb)
26
+
27
+ ---
28
+
29
+ ## Model Capabilities
30
+ The model takes both the **Writing Prompt** and the **Student's Essay** as inputs, focuses specifically on the essay body using an attention pooling layer, and grades the text across 4 core traits (scaled 0-9):
31
+ * **TA** (Task Achievement)
32
+ * **CC** (Coherence and Cohesion)
33
+ * **LR** (Lexical Resource)
34
+ * **GRA** (Grammatical Range and Accuracy)
35
+ * **OverallBand** (Calculated average rounded to the nearest 0.5)
36
+
37
+ ### Training Curves
38
+ ![Training curves](training_curves.png)
39
+
40
+ ---
41
+
42
+ ## License
43
+
44
+ This project is licensed under the MIT License.