Update README.md

README.md

@@ -14,12 +14,14 @@ tags:
 - bert
 - go_emotions
 - emotion-classification
+- sentiment-analysis
 datasets:
 - google-research-datasets/go_emotions
 metrics:
 - f1
 - precision
 - recall
+- accuracy
 widget:
 - text: I’m just chilling today.
   example_title: Neutral Example
@@ -34,125 +36,126 @@ widget:
 base_model:
 - google-bert/bert-base-uncased
 model-index:
-  - name: GoEmotions BERT Classifier
-    results:
-      - task:
-          type: multi-label-classification
-        dataset:
-          name: GoEmotions
-          type: google-research-datasets/go_emotions
-        metrics:
-          - name: Micro F1 (Optimized Thresholds)
-            type: micro-f1
-            value: 0.6006
-          - name: Macro F1
-            type: macro-f1
-            value: 0.5390
-          - name: Precision
-            type: precision
-            value: 0.5371
-          - name: Recall
-            type: recall
-            value: 0.6812
-          - name: Hamming Loss
-            type: hamming-loss
-            value: 0.0377
-          - name: Avg Positive Predictions
-            type: avg-positive-predictions
-            value: 1.4789
-      - task:
-          type: multi-label-classification
-        dataset:
-          name: GoEmotions
-          type: google-research-datasets/go_emotions
-        metrics:
-          - name: F1 (admiration)
-            type: f1
-            value: 0.6987
-          - name: F1 (amusement)
-            type: f1
-            value: 0.8071
-          - name: F1 (anger)
-            type: f1
-            value: 0.5030
-          - name: F1 (annoyance)
-            type: f1
-            value: 0.3892
-          - name: F1 (approval)
-            type: f1
-            value: 0.3915
-          - name: F1 (caring)
-            type: f1
-            value: 0.4473
-          - name: F1 (confusion)
-            type: f1
-            value: 0.4714
-          - name: F1 (curiosity)
-            type: f1
-            value: 0.5781
-          - name: F1 (desire)
-            type: f1
-            value: 0.5229
-          - name: F1 (disappointment)
-            type: f1
-            value: 0.3333
-          - name: F1 (disapproval)
-            type: f1
-            value: 0.4323
-          - name: F1 (disgust)
-            type: f1
-            value: 0.4926
-          - name: F1 (embarrassment)
-            type: f1
-            value: 0.4912
-          - name: F1 (excitement)
-            type: f1
-            value: 0.4571
-          - name: F1 (fear)
-            type: f1
-            value: 0.5860
-          - name: F1 (gratitude)
-            type: f1
-            value: 0.9102
-          - name: F1 (grief)
-            type: f1
-            value: 0.3333
-          - name: F1 (joy)
-            type: f1
-            value: 0.6135
-          - name: F1 (love)
-            type: f1
-            value: 0.8065
-          - name: F1 (nervousness)
-            type: f1
-            value: 0.4348
-          - name: F1 (optimism)
-            type: f1
-            value: 0.5564
-          - name: F1 (pride)
-            type: f1
-            value: 0.5217
-          - name: F1 (realization)
-            type: f1
-            value: 0.2513
-          - name: F1 (relief)
-            type: f1
-            value: 0.5833
-          - name: F1 (remorse)
-            type: f1
-            value: 0.6800
-          - name: F1 (sadness)
-            type: f1
-            value: 0.5570
-          - name: F1 (surprise)
-            type: f1
-            value: 0.5562
-          - name: F1 (neutral)
-            type: f1
-            value: 0.6867
-        source:
-          name: Kaggle Evaluation Notebook
-          url: https://www.kaggle.com/code/ravindranlogasanjeev/evaluation-logasanjeev-goemotions-bert/notebook
+- name: GoEmotions BERT Classifier
+  results:
+  - task:
+      type: multi-label-classification
+    dataset:
+      name: GoEmotions
+      type: google-research-datasets/go_emotions
+    metrics:
+    - name: Micro F1 (Optimized Thresholds)
+      type: micro-f1
+      value: 0.6006
+    - name: Macro F1
+      type: macro-f1
+      value: 0.539
+    - name: Precision
+      type: precision
+      value: 0.5371
+    - name: Recall
+      type: recall
+      value: 0.6812
+    - name: Hamming Loss
+      type: hamming-loss
+      value: 0.0377
+    - name: Avg Positive Predictions
+      type: avg-positive-predictions
+      value: 1.4789
+  - task:
+      type: multi-label-classification
+    dataset:
+      name: GoEmotions
+      type: google-research-datasets/go_emotions
+    metrics:
+    - name: F1 (admiration)
+      type: f1
+      value: 0.6987
+    - name: F1 (amusement)
+      type: f1
+      value: 0.8071
+    - name: F1 (anger)
+      type: f1
+      value: 0.503
+    - name: F1 (annoyance)
+      type: f1
+      value: 0.3892
+    - name: F1 (approval)
+      type: f1
+      value: 0.3915
+    - name: F1 (caring)
+      type: f1
+      value: 0.4473
+    - name: F1 (confusion)
+      type: f1
+      value: 0.4714
+    - name: F1 (curiosity)
+      type: f1
+      value: 0.5781
+    - name: F1 (desire)
+      type: f1
+      value: 0.5229
+    - name: F1 (disappointment)
+      type: f1
+      value: 0.3333
+    - name: F1 (disapproval)
+      type: f1
+      value: 0.4323
+    - name: F1 (disgust)
+      type: f1
+      value: 0.4926
+    - name: F1 (embarrassment)
+      type: f1
+      value: 0.4912
+    - name: F1 (excitement)
+      type: f1
+      value: 0.4571
+    - name: F1 (fear)
+      type: f1
+      value: 0.586
+    - name: F1 (gratitude)
+      type: f1
+      value: 0.9102
+    - name: F1 (grief)
+      type: f1
+      value: 0.3333
+    - name: F1 (joy)
+      type: f1
+      value: 0.6135
+    - name: F1 (love)
+      type: f1
+      value: 0.8065
+    - name: F1 (nervousness)
+      type: f1
+      value: 0.4348
+    - name: F1 (optimism)
+      type: f1
+      value: 0.5564
+    - name: F1 (pride)
+      type: f1
+      value: 0.5217
+    - name: F1 (realization)
+      type: f1
+      value: 0.2513
+    - name: F1 (relief)
+      type: f1
+      value: 0.5833
+    - name: F1 (remorse)
+      type: f1
+      value: 0.68
+    - name: F1 (sadness)
+      type: f1
+      value: 0.557
+    - name: F1 (surprise)
+      type: f1
+      value: 0.5562
+    - name: F1 (neutral)
+      type: f1
+      value: 0.6867
+    source:
+      name: Kaggle Evaluation Notebook
+      url: >-
+        https://www.kaggle.com/code/ravindranlogasanjeev/evaluation-logasanjeev-goemotions-bert/notebook
 ---
 
 # GoEmotions BERT Classifier
@@ -232,9 +235,6 @@ The following table shows per-class metrics on the test set using optimized thre
 #### Class-Wise F1 Scores
 ![Class-Wise F1 Scores](class_wise_f1_plot.png)
 
-#### Confusion Matrix Heatmap
-![Multi-Label Confusion Matrix Heatmap](confusion_matrix_heatmap.png)
-
 #### Training Curves
 ![Training and Validation Loss and Micro F1](training_curves_plot.png)