{
  "present": true,
  "run_name": "baseline-03",
  "config": {
    "epochs": 10,
    "batch_size": 64,
    "learning_rate": 0.001,
    "weight_decay": 0.0001,
    "image_size": 64,
    "val_frac": 0.15,
    "augment": true,
    "seed": 42
  },
  "history": [
    {
      "epoch": 1,
      "train_loss": 0.479377065386091,
      "train_acc": 0.7932291655313402,
      "val_loss": 0.3618093580007553,
      "val_acc": 0.8224999979138374,
      "seconds": 2.5827352919150144
    },
    {
      "epoch": 2,
      "train_loss": 0.3089326300791332,
      "train_acc": 0.838616071712403,
      "val_loss": 0.29158738628029823,
      "val_acc": 0.8389843702316284,
      "seconds": 2.485130582936108
    },
    {
      "epoch": 3,
      "train_loss": 0.2782897197064899,
      "train_acc": 0.8564732131503877,
      "val_loss": 0.2897236980497837,
      "val_acc": 0.8520312532782555,
      "seconds": 2.4839513751212507
    },
    {
      "epoch": 4,
      "train_loss": 0.2619182669690677,
      "train_acc": 0.8662202415012178,
      "val_loss": 0.2705635577440262,
      "val_acc": 0.8640625029802322,
      "seconds": 2.4814148750156164
    },
    {
      "epoch": 5,
      "train_loss": 0.25110251598414923,
      "train_acc": 0.8723214325450716,
      "val_loss": 0.2844527140259743,
      "val_acc": 0.8592187464237213,
      "seconds": 2.4968186249025166
    },
    {
      "epoch": 6,
      "train_loss": 0.23687684216669627,
      "train_acc": 0.8831101201829457,
      "val_loss": 0.25770158506929874,
      "val_acc": 0.8754687458276749,
      "seconds": 2.4867920419201255
    },
    {
      "epoch": 7,
      "train_loss": 0.23243631564435505,
      "train_acc": 0.8853422644592467,
      "val_loss": 0.25699080526828766,
      "val_acc": 0.8745312392711639,
      "seconds": 2.5252511671278626
    },
    {
      "epoch": 8,
      "train_loss": 0.22684153346788316,
      "train_acc": 0.8871279812994457,
      "val_loss": 0.24439864233136177,
      "val_acc": 0.8792187497019768,
      "seconds": 2.489842291921377
    },
    {
      "epoch": 9,
      "train_loss": 0.21713796293451673,
      "train_acc": 0.8968005946704319,
      "val_loss": 0.2730095852166414,
      "val_acc": 0.8478906229138374,
      "seconds": 2.485344582935795
    },
    {
      "epoch": 10,
      "train_loss": 0.2112113730538459,
      "train_acc": 0.8970238112267994,
      "val_loss": 0.2408575601875782,
      "val_acc": 0.8744531273841858,
      "seconds": 2.4936056670267135
    }
  ],
  "total_seconds": 25.011042000027373,
  "best_val_acc": 0.8792187497019768,
  "n_train": 2656,
  "n_val": 468,
  "eval": {
    "per_class": [
      {
        "label": "ok",
        "tp": 186,
        "fp": 62,
        "fn": 44,
        "tn": 176,
        "precision": 0.75,
        "recall": 0.808695652173913,
        "f1": 0.7782426778242677
      },
      {
        "label": "crack",
        "tp": 123,
        "fp": 69,
        "fn": 71,
        "tn": 205,
        "precision": 0.640625,
        "recall": 0.634020618556701,
        "f1": 0.6373056994818652
      },
      {
        "label": "burn",
        "tp": 23,
        "fp": 13,
        "fn": 15,
        "tn": 417,
        "precision": 0.6388888888888888,
        "recall": 0.6052631578947368,
        "f1": 0.6216216216216216
      },
      {
        "label": "broken-wire",
        "tp": 45,
        "fp": 0,
        "fn": 1,
        "tn": 422,
        "precision": 1.0,
        "recall": 0.9782608695652174,
        "f1": 0.989010989010989
      },
      {
        "label": "short-circuit",
        "tp": 25,
        "fp": 9,
        "fn": 11,
        "tn": 423,
        "precision": 0.7352941176470589,
        "recall": 0.6944444444444444,
        "f1": 0.7142857142857144
      }
    ],
    "macro_f1": 0.7480933404448915,
    "element_acc": 0.8739316239316239,
    "exact_match_acc": 0.6452991452991453,
    "n_samples": 468,
    "threshold": 0.5
  },
  "samples": [
    {
      "image": "data/p5_samples/sample_00.png",
      "ground_truth": {
        "severity": 0,
        "probability": 0.0,
        "module_type": "mono"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 96.0
          },
          {
            "label": "crack",
            "similarity_pct": 95.88
          }
        ]
      }
    },
    {
      "image": "data/p5_samples/sample_01.png",
      "ground_truth": {
        "severity": 1,
        "probability": 0.3333333333333333,
        "module_type": "mono"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 94.56
          },
          {
            "label": "crack",
            "similarity_pct": 94.43
          }
        ]
      }
    },
    {
      "image": "data/p5_samples/sample_02.png",
      "ground_truth": {
        "severity": 2,
        "probability": 0.6666666666666666,
        "module_type": "poly"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 98.25
          },
          {
            "label": "crack",
            "similarity_pct": 98.17
          }
        ]
      }
    },
    {
      "image": "data/p5_samples/sample_03.png",
      "ground_truth": {
        "severity": 3,
        "probability": 1.0,
        "module_type": "mono"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 99.65
          },
          {
            "label": "crack",
            "similarity_pct": 99.43
          }
        ]
      }
    },
    {
      "image": "data/p5_samples/sample_04.png",
      "ground_truth": {
        "severity": 3,
        "probability": 1.0,
        "module_type": "mono"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 99.84
          },
          {
            "label": "crack",
            "similarity_pct": 99.79
          }
        ]
      }
    },
    {
      "image": "data/p5_samples/sample_05.png",
      "ground_truth": {
        "severity": 3,
        "probability": 1.0,
        "module_type": "mono"
      },
      "predicted": {
        "verdict": "OK",
        "top": [
          {
            "label": "ok",
            "similarity_pct": 99.27
          },
          {
            "label": "crack",
            "similarity_pct": 99.12
          }
        ]
      }
    }
  ]
}