deploy_config.json

{
  "model": "tf_efficientnet_b0",
  "class_names": [
    "illustration",
    "painting_physical",
    "real_photo",
    "rendered_2d",
    "rendered_3d",
    "screenshot"
  ],
  "merge": {
    "anchor": "checkpoints/v2.70_clean_core_v263_finish_weighted8_lr5e5/best_model.pth",
    "donor": "checkpoints/v2.29_res244_4plus4/best_model.pth",
    "alpha": 0.1
  },
  "class_indices": {
    "illustration": 0,
    "painting_physical": 1,
    "real_photo": 2,
    "rendered_2d": 3,
    "rendered_3d": 4,
    "screenshot": 5
  },
  "input_size": 336,
  "eval_size": 336,
  "normalization": {
    "mean": [
      0.485,
      0.456,
      0.406
    ],
    "std": [
      0.229,
      0.224,
      0.225
    ]
  },
  "checkpoint_file": "best_model.safetensors",
  "holdout_metrics": {
    "accuracy": 0.9140893470790378,
    "macro_f1": 0.9134163980072622,
    "harmonic_f1": 0.9123984895680628,
    "per_class_f1": {
      "illustration": 0.8988764044943819,
      "painting_physical": 0.9504950495049505,
      "real_photo": 0.923076923076923,
      "rendered_2d": 0.8936170212765958,
      "rendered_3d": 0.8659793814432989,
      "screenshot": 0.9484536082474226
    },
    "per_class_recall": {
      "illustration": 0.975609756097561,
      "painting_physical": 0.96,
      "real_photo": 0.96,
      "rendered_2d": 0.84,
      "rendered_3d": 0.84,
      "screenshot": 0.92
    },
    "min_recall": 0.84,
    "bottom_2_mean_recall": 0.84,
    "confusion_matrix": [
      [
        40,
        1,
        0,
        0,
        0,
        0
      ],
      [
        0,
        48,
        1,
        0,
        1,
        0
      ],
      [
        1,
        1,
        48,
        0,
        0,
        0
      ],
      [
        4,
        0,
        0,
        42,
        3,
        1
      ],
      [
        2,
        0,
        4,
        2,
        42,
        0
      ],
      [
        1,
        1,
        1,
        0,
        1,
        46
      ]
    ],
    "worst_confusion_pairs": [
      {
        "from": "rendered_2d",
        "to": "illustration",
        "rate": 0.08,
        "count": 4
      },
      {
        "from": "rendered_3d",
        "to": "real_photo",
        "rate": 0.08,
        "count": 4
      },
      {
        "from": "rendered_2d",
        "to": "rendered_3d",
        "rate": 0.06,
        "count": 3
      },
      {
        "from": "rendered_3d",
        "to": "illustration",
        "rate": 0.04,
        "count": 2
      },
      {
        "from": "rendered_3d",
        "to": "rendered_2d",
        "rate": 0.04,
        "count": 2
      },
      {
        "from": "illustration",
        "to": "painting_physical",
        "rate": 0.024390243902439025,
        "count": 1
      },
      {
        "from": "painting_physical",
        "to": "real_photo",
        "rate": 0.02,
        "count": 1
      },
      {
        "from": "painting_physical",
        "to": "rendered_3d",
        "rate": 0.02,
        "count": 1
      },
      {
        "from": "real_photo",
        "to": "illustration",
        "rate": 0.02,
        "count": 1
      },
      {
        "from": "real_photo",
        "to": "painting_physical",
        "rate": 0.02,
        "count": 1
      }
    ]
  }
}