Update README.md

README.md

@@ -1,39 +1,38 @@
 ---
 library_name: transformers
-license: mit
+license: cc-by-4.0
 base_model: roberta-base
 tags:
 - generated_from_trainer
+- text-classification
+- classification
+- nlp
+- vulnerability
+- CWE
 metrics:
 - accuracy
 model-index:
 - name: cwe-parent-vulnerability-classification-roberta-base
   results: []
+datasets:
+- CIRCL/vulnerability-cwe-patch
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# cwe-parent-vulnerability-classification-roberta-base
+# CWE guessing
+
+This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on the
+[CIRCL/vulnerability-cwe-patch](https://huggingface.co/datasets/CIRCL/vulnerability-cwe-patch) dataset.
+
+The goal is to predict CWE categories from Git commit messages and vulnerability descriptions. Predicted child CWEs are mapped to their parent CWEs if applicable.
 
-This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
 - Loss: 1.7510
 - Accuracy: 0.5455
 - F1 Macro: 0.3776
 
-## Model description
-
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
 ## Training procedure
 
 ### Training hyperparameters