PyTorch
medical-imaging
ultrasound
thyroid
classification
resnet
ml-intern
File size: 965 Bytes
301e62a
653a862
 
301e62a
653a862
 
301e62a
653a862
 
301e62a
653a862
301e62a
653a862
 
301e62a
653a862
301e62a
653a862
 
301e62a
653a862
301e62a
653a862
 
301e62a
653a862
301e62a
653a862
 
301e62a
653a862
 
 
 
301e62a
 
653a862
 
 
 
 
 
 
3fe7874
301e62a
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
{
  "best_epoch": 12,
  "best_val_auc": 0.9318691478479817,
  "test_metrics": {
    "accuracy": 0.8891108891108891,
    "sensitivity": 0.9174825174825175,
    "sensitivity_ci": [
      0.8948475523330561,
      0.9365925537099365
    ],
    "specificity": 0.8181818181818182,
    "specificity_ci": [
      0.7685054712783398,
      0.8611264590466025
    ],
    "ppv": 0.9265536723163842,
    "ppv_ci": [
      0.9047977037903637,
      0.9446634522191668
    ],
    "npv": 0.7986348122866894,
    "npv_ci": [
      0.7480838301457988,
      0.8430321946896866
    ],
    "auc": 0.9312704777739742,
    "auc_ci": [
      0.9124902286540131,
      0.9482509242450335
    ],
    "tp": 656,
    "tn": 234,
    "fp": 52,
    "fn": 59
  },
  "config": {
    "backbone": "resnet18.a1_in1k",
    "img_size": 128,
    "batch_size": 16,
    "epochs_trained": 32,
    "lr": 0.0001,
    "weight_decay": 0.05,
    "normalization": "none",
    "optimization_metric": "auc"
  }
}