Update README.md

README.md

@@ -5,10 +5,10 @@ license: afl-3.0
 This text categorization model can predict if a word sequence is suicidal (1) or not (0).
 
 ## Data
-The model was trained on the [Suicide and Depression Dataset](https://www.kaggle.com/nikhileswarkomati/suicide-watch) obtained from Kaggle. The dataset was taken from Reddit and contains 232,074 data divided into two categories: suicide and non-suicide.
+The model was trained on the [Suicide and Depression Dataset](https://www.kaggle.com/) obtained from Kaggle. The dataset was taken from Reddit and contains 232,074 data divided into two categories: suicide and non-suicide.
 
 ## Parameters
-The model fine-tuning was conducted on 1 epoch, with batch size of 6, and learning rate of 0.00001. Due to limited computing resources and time, we were unable to scale up the number of epochs and batch size.
+The model fine-tuning was conducted on 1 epoch, with a batch size of 6, and a learning rate of 0.00001. Due to limited computing resources and time, we were unable to scale up the number of epochs and batch size.
 
 ## Performance
 Following fine-tuning on the mentioned dataset, the model generated the subsequent results: