Update README.md

README.md

@@ -3,6 +3,7 @@ language:
 - en
 tags:
 - Text Classification
+co2_eq_emissions: 0.319355 Kg
 widget:
 - text: "Nevertheless, Trump and other Republicans have tarred the protests as havens for terrorists intent on destroying property."
   example_title: "Biased example 1"
@@ -14,21 +15,18 @@ widget:
   example_title: "Non-Biased example 2"
 ---
 
-## This model is part of the Research topic "Bias and Fairness in AI" conducted by Shaina Raza, Deepak John Reji, Chen Ding
+## About the Model
+This model is trained to detect bias in a sentence. 
 
 - Dataset : MBAD Data
-- datasize : 17775 entries
-- Label distribution : Biased - 10651, Non-Biased - 7124
-- Train-Test split : 90 : 10
-- tokenizer : distilbert-base-uncased
-- model : distilbert-base-uncased
-- optimizer : adam
-- Learning rate : 5e-5
-- Model parameters : 66,955,010
-- epochs : 30
+- Carbon emission 0.319355 Kg
+
 - Train accuracy : 0.7697
 - Test accuracy : 0.62
 - Train loss : 0.4506
 - Test loss : 0.9644
-- Carbon emission 0.319355 Kg
+
+## Author
+This model is part of the Research topic "Bias and Fairness in AI" conducted by Shaina Raza, Deepak John Reji, Chen Ding. If you use this work (code, model or dataset), please cite as:
+> Bias & Fairness in AI, (2020), GitHub repository, <https://github.com/dreji18/Fairness-in-AI/tree/dev>