Update README.md

README.md

@@ -9,14 +9,14 @@ metrics:
 - f1
 - accuracy
 model-index:
-- name: our_data
+- name: Cyber-ThreaD/SecureBERT-AttackER
   results: []
 ---
 
 <!-- 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. -->
 
-# our_data
+# Cyber-ThreaD/SecureBERT-AttackER
 
 This model is a fine-tuned version of [ehsanaghaei/SecureBERT](https://huggingface.co/ehsanaghaei/SecureBERT) on an unknown dataset.
 It achieves the following results on the evaluation set:
@@ -87,3 +87,21 @@ The following hyperparameters were used during training:
 - Pytorch 2.1.0+cu118
 - Datasets 2.15.0
 - Tokenizers 0.15.0
+
+
+
+### Citing & Authors
+
+If you use the model kindly cite the following work
+
+```
+@inproceedings{deka2024attacker,
+  title={AttackER: Towards Enhancing Cyber-Attack Attribution with a Named Entity Recognition Dataset},
+  author={Deka, Pritam and Rajapaksha, Sampath and Rani, Ruby and Almutairi, Amirah and Karafili, Erisa},
+  booktitle={International Conference on Web Information Systems Engineering},
+  pages={255--270},
+  year={2024},
+  organization={Springer}
+}
+
+```