{
  "parameters": 12040967,
  "selected_epoch": 2,
  "validation": {
    "n": 1232,
    "accuracy": 0.726461038961039,
    "macro_f1": 0.7323809921926641,
    "weighted_f1": 0.7245863959383607,
    "per_class": {
      "ANGRY": {
        "precision": 0.7840909090909091,
        "recall": 0.7419354838709677,
        "f1-score": 0.7624309392265194,
        "support": 186.0
      },
      "FEAR": {
        "precision": 0.8203592814371258,
        "recall": 0.9013157894736842,
        "f1-score": 0.8589341692789969,
        "support": 152.0
      },
      "HAPPY": {
        "precision": 0.6979166666666666,
        "recall": 0.5317460317460317,
        "f1-score": 0.6036036036036037,
        "support": 126.0
      },
      "HATE": {
        "precision": 0.8207547169811321,
        "recall": 0.8446601941747572,
        "f1-score": 0.8325358851674641,
        "support": 103.0
      },
      "SAD": {
        "precision": 0.6287128712871287,
        "recall": 0.7055555555555556,
        "f1-score": 0.6649214659685864,
        "support": 180.0
      },
      "SURPRISE": {
        "precision": 0.6706586826347305,
        "recall": 0.7567567567567568,
        "f1-score": 0.7111111111111111,
        "support": 148.0
      },
      "OTHER": {
        "precision": 0.7138364779874213,
        "recall": 0.6735905044510386,
        "f1-score": 0.6931297709923664,
        "support": 337.0
      },
      "accuracy": 0.726461038961039,
      "macro avg": {
        "precision": 0.7337613722978735,
        "recall": 0.7365086165755417,
        "f1-score": 0.7323809921926641,
        "support": 1232.0
      },
      "weighted avg": {
        "precision": 0.7272722813813256,
        "recall": 0.726461038961039,
        "f1-score": 0.7245863959383607,
        "support": 1232.0
      }
    },
    "confusion_matrix": [
      [
        138,
        2,
        1,
        6,
        12,
        9,
        18
      ],
      [
        1,
        137,
        2,
        0,
        3,
        4,
        5
      ],
      [
        4,
        2,
        67,
        2,
        22,
        7,
        22
      ],
      [
        5,
        0,
        1,
        87,
        2,
        2,
        6
      ],
      [
        7,
        5,
        2,
        6,
        127,
        7,
        26
      ],
      [
        7,
        5,
        3,
        1,
        6,
        112,
        14
      ],
      [
        14,
        16,
        20,
        4,
        30,
        26,
        227
      ]
    ],
    "label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE",
      "OTHER"
    ],
    "f1_label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE",
      "OTHER"
    ]
  },
  "test": {
    "n": 1151,
    "accuracy": 0.7002606429192006,
    "macro_f1": 0.6916643550452649,
    "weighted_f1": 0.7020107747970785,
    "per_class": {
      "ANGRY": {
        "precision": 0.75,
        "recall": 0.6428571428571429,
        "f1-score": 0.6923076923076923,
        "support": 154.0
      },
      "FEAR": {
        "precision": 0.6825396825396826,
        "recall": 0.7543859649122807,
        "f1-score": 0.7166666666666667,
        "support": 57.0
      },
      "HAPPY": {
        "precision": 0.8730964467005076,
        "recall": 0.6254545454545455,
        "f1-score": 0.7288135593220338,
        "support": 275.0
      },
      "HATE": {
        "precision": 0.7254901960784313,
        "recall": 0.5692307692307692,
        "f1-score": 0.6379310344827587,
        "support": 65.0
      },
      "SAD": {
        "precision": 0.6949685534591195,
        "recall": 0.8435114503816794,
        "f1-score": 0.7620689655172413,
        "support": 262.0
      },
      "SURPRISE": {
        "precision": 0.8235294117647058,
        "recall": 0.5793103448275863,
        "f1-score": 0.680161943319838,
        "support": 145.0
      },
      "OTHER": {
        "precision": 0.5208333333333334,
        "recall": 0.7772020725388601,
        "f1-score": 0.6237006237006237,
        "support": 193.0
      },
      "accuracy": 0.7002606429192006,
      "macro avg": {
        "precision": 0.7243510891251114,
        "recall": 0.6845646128861234,
        "f1-score": 0.6916643550452649,
        "support": 1151.0
      },
      "weighted avg": {
        "precision": 0.7329952272267632,
        "recall": 0.7002606429192006,
        "f1-score": 0.7020107747970785,
        "support": 1151.0
      }
    },
    "confusion_matrix": [
      [
        99,
        5,
        2,
        2,
        14,
        6,
        26
      ],
      [
        1,
        43,
        1,
        0,
        9,
        1,
        2
      ],
      [
        5,
        4,
        172,
        3,
        27,
        6,
        58
      ],
      [
        8,
        3,
        0,
        37,
        8,
        2,
        7
      ],
      [
        3,
        4,
        9,
        5,
        221,
        1,
        19
      ],
      [
        8,
        1,
        7,
        1,
        18,
        84,
        26
      ],
      [
        8,
        3,
        6,
        3,
        21,
        2,
        150
      ]
    ],
    "label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE",
      "OTHER"
    ],
    "f1_label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE",
      "OTHER"
    ]
  },
  "six_emotion_subset": {
    "n": 958,
    "accuracy": 0.6847599164926931,
    "macro_f1": 0.7194582654221393,
    "weighted_f1": 0.7345959444551162,
    "per_class": {
      "ANGRY": {
        "precision": 0.7983870967741935,
        "recall": 0.6428571428571429,
        "f1-score": 0.7122302158273381,
        "support": 154.0
      },
      "FEAR": {
        "precision": 0.7166666666666667,
        "recall": 0.7543859649122807,
        "f1-score": 0.7350427350427351,
        "support": 57.0
      },
      "HAPPY": {
        "precision": 0.900523560209424,
        "recall": 0.6254545454545455,
        "f1-score": 0.7381974248927039,
        "support": 275.0
      },
      "HATE": {
        "precision": 0.7708333333333334,
        "recall": 0.5692307692307692,
        "f1-score": 0.6548672566371682,
        "support": 65.0
      },
      "SAD": {
        "precision": 0.7441077441077442,
        "recall": 0.8435114503816794,
        "f1-score": 0.7906976744186046,
        "support": 262.0
      },
      "SURPRISE": {
        "precision": 0.84,
        "recall": 0.5793103448275863,
        "f1-score": 0.6857142857142857,
        "support": 145.0
      },
      "micro avg": {
        "precision": 0.8,
        "recall": 0.6847599164926931,
        "f1-score": 0.7379077615298087,
        "support": 958.0
      },
      "macro avg": {
        "precision": 0.7950864001818937,
        "recall": 0.6691250362773339,
        "f1-score": 0.7194582654221393,
        "support": 958.0
      },
      "weighted avg": {
        "precision": 0.8124279619871743,
        "recall": 0.6847599164926931,
        "f1-score": 0.7345959444551162,
        "support": 958.0
      }
    },
    "confusion_matrix": [
      [
        99,
        5,
        2,
        2,
        14,
        6,
        26
      ],
      [
        1,
        43,
        1,
        0,
        9,
        1,
        2
      ],
      [
        5,
        4,
        172,
        3,
        27,
        6,
        58
      ],
      [
        8,
        3,
        0,
        37,
        8,
        2,
        7
      ],
      [
        3,
        4,
        9,
        5,
        221,
        1,
        19
      ],
      [
        8,
        1,
        7,
        1,
        18,
        84,
        26
      ],
      [
        0,
        0,
        0,
        0,
        0,
        0,
        0
      ]
    ],
    "label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE",
      "OTHER"
    ],
    "f1_label_order": [
      "ANGRY",
      "FEAR",
      "HAPPY",
      "HATE",
      "SAD",
      "SURPRISE"
    ]
  },
  "original_checkpoint_sha256": "9fc3e1d865d7177a0f2a3c5d90e2114357b355b98ac0b80292a7d9d5e9a69d38"
}