Update README.md

README.md

@@ -37,27 +37,19 @@ probably proofread and complete it, then remove this comment. -->
 
 # nathanReitinger/mlcb
 
-This model is a fine-tuned version of [dbernsohn/roberta-javascript](https://huggingface.co/dbernsohn/roberta-javascript) on an unknown dataset.
+This model is a fine-tuned version of [dbernsohn/roberta-javascript](https://huggingface.co/dbernsohn/roberta-javascript) on the [mlcb dataset](https://huggingface.co/datasets/nathanReitinger/mlcb).
 It achieves the following results on the evaluation set:
 - Train Loss: 0.0463
 - Validation Loss: 0.0930
 - Train Accuracy: 0.9708
 - Epoch: 4
 
-## Model description
-
-More information needed
-
 ## Intended uses & limitations
 
-More information needed
+The model can be used to identify whether a JavaScript program is engaging in canvas fingerprinting. 
 
 ## Training and evaluation data
 
-More information needed
-
-## Training procedure
-
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
@@ -81,3 +73,19 @@ The following hyperparameters were used during training:
 - TensorFlow 2.11.0
 - Datasets 2.13.2
 - Tokenizers 0.13.3
+
+# Citation 
+
+@inproceedings{reitinger2021ml,
+  title={ML-CB: Machine Learning Canvas Block.},
+  author={Nathan Reitinger and Michelle L Mazurek},
+  journal={Proc.\ PETS},
+  volume={2021},
+  number={3},
+  pages={453--473},
+  year={2021}
+}
+
+- [OSF] (https://osf.io/shbe7/)
+- [GitHub] (https://github.com/SP2-MC2/ML-CB)
+- [Data] (https://dataverse.harvard.edu/dataverse/ml-cb)