Hugging Face's logo

iraqigold
/
Medical-Modality-Classifier

Model card Files
xet
 Community
1
Medical-Modality-Classifier
File size: 1,277 Bytes
435f28b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
{
  "config": {
    "data_dir": null,
    "output_dir": "/content/training_runs/modality_classifier",
    "hf_token": "***redacted***",
    "hf_dataset_repo": "iraqigold/Medical-Modality-Dataset",
    "hf_model_repo": "iraqigold/Medical-Modality-Classifier",
    "download_hf": false,
    "upload_to_hf": true,
    "backbone": "resnet18",
    "epochs": 15,
    "batch_size": 32,
    "num_workers": 0,
    "lr": 0.0001,
    "weight_decay": 0.0001,
    "val_ratio": 0.15,
    "test_ratio": 0.1,
    "seed": 42,
    "save_every": 3,
    "upload_every_epochs": 3,
    "resume_training": true,
    "force_balance": true
  },
  "classes": [
    "CT",
    "MRI",
    "US",
    "XRAY"
  ],
  "class_mapping": {
    "CT": 0,
    "MRI": 1,
    "US": 2,
    "XRAY": 3
  },
  "train_samples": 6000,
  "val_samples": 1200,
  "test_samples": 800,
  "best_val_accuracy": 0.9991666666666666,
  "test_metrics": {
    "test_loss": 0.0007467886476661079,
    "test_accuracy": 1.0,
    "per_class_accuracy": {
      "CT": 1.0,
      "MRI": 1.0,
      "US": 1.0,
      "XRAY": 1.0
    },
    "per_class_correct": {
      "CT": 206,
      "MRI": 201,
      "US": 178,
      "XRAY": 215
    },
    "per_class_total": {
      "CT": 206,
      "MRI": 201,
      "US": 178,
      "XRAY": 215
    }
  }
}