Update README.md

README.md

@@ -2,11 +2,17 @@
 license: apache-2.0
 tags:
 - generated_from_trainer
+- code
 metrics:
 - accuracy
 model-index:
 - name: moderate_severe_depression_model
   results: []
+datasets:
+- christinacdl/balanced_depression_dataset
+language:
+- en
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -20,6 +26,24 @@ It achieves the following results on the evaluation set:
 - Macro F1: 0.8927
 - Accuracy: 0.8932
 
+Results on Test set:
+
+-Accuracy: 0.8817204301075269
+
+-F1 score: 0.8819253137510324 
+
+-Precision: 0.8855477220587717
+
+-Recall : 0.8817204301075269
+
+-Matthews Correlation Coefficient: 0.8242972089300715
+
+-Precision of each class: [0.98420129 0.85636693 0.81607495]
+
+-Recall of each class: [0.93548387 0.78921023 0.92046719]
+
+-F1 score of each class: [0.95922441 0.82141823 0.8651333 ]
+
 ## Model description
 
 More information needed
@@ -63,4 +87,4 @@ The following hyperparameters were used during training:
 - Transformers 4.27.1
 - Pytorch 2.0.1+cu118
 - Datasets 2.9.0
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3