Update README.md

README.md

@@ -3,23 +3,23 @@ license: apache-2.0
 language:
 - en
 ---
-# Toxicity_model
+# Toxicity-classifier
 
-The Toxicity_model is used to differentiate not toxic from toxic responses.
+The toxicity classifier is used to differentiate between non-toxic and toxic comments.
 
-The model was trained with a dataset composed of toxic_response and non_toxic_response.
+The model was trained with a dataset composed of toxic and non-toxic comments extracted from web forums.
 
 ## Details
-- Size: 4,689,681 parameters
-- Dataset: [Toxic Comment Classification Challenge Dataset](https://github.com/tianqwang/Toxic-Comment-Classification-Challenge)
-- Language: English
-- Number of Training Steps: 20
-- Batch size: 16
-- Optimizer: Adam
-- Learning Rate: 0.001
-- GPU: T4
-- This repository has the source [code used](https://github.com/Nkluge-correa/teeny-tiny_castle/blob/master/ML%20Intro%20Course/15_toxicity_detection.ipynb) to train this model.
-
+- **Size:** 4,689,681 parameters
+- **Dataset:** [Toxic Comment Classification Challenge Dataset](https://huggingface.co/datasets/AiresPucrs/toxic-comments)
+- **Language:** English
+- **Number of Epochs:** 20
+- **Batch Size:** 16
+- **Optimizer:** Adam
+- **Learning Rate:** 0.001
+- **Hardware:** Tesla V4 
+- **Emissions:** Not measured
+- **Total Energy Consumption:** Not measured
 ## Usage
 
 ⚠️ THE EXAMPLES BELOW CONTAIN TOXIC/OFFENSIVE LANGUAGE ⚠️
@@ -65,6 +65,7 @@ __________________________________________________
 ```
 
 # Cite as 🤗
+
 ```
 @misc{teenytinycastle,
     doi = {10.5281/zenodo.7112065},