| { | |
| "model_name": "convnext_tiny", | |
| "num_classes": 2, | |
| "image_size": 384, | |
| "id2label": { | |
| "0": "good", | |
| "1": "bad" | |
| }, | |
| "label2id": { | |
| "good": 0, | |
| "bad": 1 | |
| }, | |
| "threshold": 0.5, | |
| "version": "0.2.0", | |
| "train_ckpt_epoch": 7, | |
| "dtype": "float32", | |
| "train_dataset": "v1.0.4_768", | |
| "metrics_holdout_v0": { | |
| "accuracy": 0.94140625, | |
| "precision_bad": 0.9819277108433735, | |
| "recall_bad": 0.9314285714285714, | |
| "f1_bad": 0.9560117302052786, | |
| "false_bad_rate": 0.03703703703703709, | |
| "auc": 0.9841644620811287, | |
| "confusion": { | |
| "tp": 652, | |
| "fp": 12, | |
| "tn": 312, | |
| "fn": 48 | |
| }, | |
| "n": 1024 | |
| }, | |
| "onnx": { | |
| "filename": "model.onnx", | |
| "opset": 17, | |
| "input_name": "pixel_values", | |
| "output_name": "logits", | |
| "input_shape": [ | |
| null, | |
| 3, | |
| 384, | |
| 384 | |
| ] | |
| } | |
| } | |