training_metadata.json

{
  "training_config": {
    "model_name": "prajjwal1/bert-tiny",
    "max_length": 128,
    "batch_size": 32,
    "learning_rate": 3e-05,
    "num_epochs": 15,
    "patience": 3,
    "test_size": 0.2,
    "device": "cpu",
    "csv_file": "all_EN_calls.csv",
    "s3_bucket": "voicex-call-recordings"
  },
  "final_metrics": {
    "accuracy": 0.9774647887323944,
    "precision": 0.9578947368421052,
    "recall": 0.9578947368421052,
    "f1": 0.9578947368421052,
    "confusion_matrix": [
      [
        512,
        8
      ],
      [
        8,
        182
      ]
    ]
  },
  "pos_weight": 2.729303547963206,
  "threshold": 0.5,
  "training_history": {
    "train_losses": [
      0.9435733710781912,
      0.6628189873829317,
      0.40206739406907155,
      0.28053958831208475,
      0.21479346770583913,
      0.180794070108553,
      0.14911148521337617,
      0.13325696530636777,
      0.12835281459468134,
      0.11012767288792,
      0.10539512767383222,
      0.09656323011169272
    ],
    "val_losses": [
      0.8112381230229917,
      0.4864982029666071,
      0.34563232180864917,
      0.26932784072730853,
      0.24466017180162927,
      0.2034845212879388,
      0.1938699973018273,
      0.19390630900211955,
      0.1721272283922071,
      0.17268858526064002,
      0.17224800457125125,
      0.18237287065257196
    ],
    "val_accuracies": [
      0.9042253521126761,
      0.9690140845070423,
      0.9704225352112676,
      0.971830985915493,
      0.9690140845070423,
      0.976056338028169,
      0.9746478873239437,
      0.9746478873239437,
      0.9774647887323944,
      0.9732394366197183,
      0.9746478873239437,
      0.9774647887323944
    ]
  }
}