Update README.md

README.md

@@ -14,7 +14,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 # polibert-malaysia-ver4
 
-This model is a fine-tuned version of [bert-base-uncased](https://huggingface.co/bert-base-uncased) on an unknown dataset.
+This model is new version of YagiASAFAS/polibert-malaysia-ver2.
+What is new is that this model used a new dataset which not only used tnwei/ms-newspapers dataset but also almost 10k of instagram posts regarding several topics about Malaysia.
+By doing so, this model captures not only formal sentences such as News, but also captures informal sentences such as personal posts.
+As a tradeoff, the accuracy was quite lower compared to the previous one(ver3).
+This model is an updated version of the ver3, increasing accuracy.
 It achieves the following results on the evaluation set:
 - Loss: 0.2641
 - Accuracy: 0.9371
@@ -45,6 +49,19 @@ The following hyperparameters were used during training:
 - num_epochs: 8
 - mixed_precision_training: Native AMP
 
+### Label Mappings
+- 0: Economic Concerns
+- 1: Racial discrimination or polarization
+- 2: Leadership weaknesses
+- 3: Development and infrastructure gaps
+- 4: Corruption
+- 5: Political instablility
+- 6: Socials and Public safety
+- 7: Administration
+- 8: Education
+- 9: Religion issues
+- 10: Environmental
+
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |