Update README.md

README.md

@@ -36,13 +36,20 @@ library_name: transformers
 
 # roberta-base-suicide-prediction-phr
 
-This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on a Suicide Prediction dataset sourced from Reddit. 
+This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on this [dataset](https://huggingface.co/datasets/vibhorag101/suicide_prediction_dataset_phr)   sourced from Reddit. 
 It achieves the following results on the evaluation/validation set:
 - Loss: 0.1543
-- Accuracy: {'accuracy': 0.9652972367116438}
-- Recall: {'recall': 0.966571403827834}
-- Precision: {'precision': 0.9638169257340242}
-- F1: {'f1': 0.9651921995935487}
+- Accuracy: 0.9652972367116438
+- Recall: 0.966571403827834
+- Precision: 0.9638169257340242
+- F1: 0.9651921995935487
+
+It achieves the following result on validation partition of this updated [dataset](vibhorag101/phr_suicide_prediction_dataset_clean_light)
+- Loss: 0.08761
+- Accuracy: 0.97065
+- Recall: 0.96652
+- Precision: 0.97732
+- F1: 0.97189
 
 ## Model description
 This model is a finetune of roberta-base to detect suicidal tendencies in a given text.