Skip to content

Latest commit

 

History

History
77 lines (65 loc) · 3.28 KB

README.md

File metadata and controls

77 lines (65 loc) · 3.28 KB

Chest X Ray Classification

Under construction

Run

configure basic parameters in settings.yml

python train.py

Test Results

Model Pos Weight AUROC Accuracy Precision Recall F1 Score
DenseNet121 No 0.8777 0.8227 0.7115 0.6066 0.6549
Yes 0.8745 0.8364 0.6964 0.6724 0.6842
VGG19_bn No 0.6296 0.7500 0.5405 0.3448 0.4211
Yes 0.5821 0.2636 0.2636 1.0000 0.4173
ResNet50 No 0.9043 0.8409 0.7091 0.6724 0.6903
Yes 0.8631 0.8136 0.6735 0.5690 0.6168
3-layer CNN No 0.6853 0.7864 0.7895 0.2586 0.3896
Yes 0.6672 0.2636 0.2636 1.0000 0.4173
EfficientNetB7 No 0.8992 0.8136 0.6133 0.7931 0.6917
Yes 0.9215 0.8591 0.7368 0.7241 0.7304