TN5000_model / metrics_summary.json
Johnyquest7's picture
End of training
94a12ca verified
{
"model": "microsoft/swinv2-base-patch4-window8-256",
"dataset": "TN5000",
"validation": {
"loss": 0.12442023307085037,
"accuracy": 0.8785714285714286,
"f1": 0.9126413155190134,
"sensitivity": 0.888,
"specificity": 0.855,
"ppv": 0.9386892177589852,
"npv": 0.7533039647577092,
"auc_roc": 0.93988,
"tp": 444,
"tn": 171,
"fp": 29,
"fn": 56,
"runtime": 4.4047,
"samples_per_second": 158.921,
"steps_per_second": 9.989
},
"test": {
"loss": 0.12932181358337402,
"accuracy": 0.872,
"f1": 0.9080459770114943,
"sensitivity": 0.8745387453874539,
"specificity": 0.8653846153846154,
"ppv": 0.9442231075697212,
"npv": 0.7258064516129032,
"auc_roc": 0.9366839341470338,
"tp": 474,
"tn": 180,
"fp": 28,
"fn": 68,
"runtime": 4.6692,
"samples_per_second": 160.627,
"steps_per_second": 10.066
},
"confusion_matrix_validation": {
"tp": 444,
"tn": 171,
"fp": 29,
"fn": 56
},
"confusion_matrix_test": {
"tp": 474,
"tn": 180,
"fp": 28,
"fn": 68
},
"class_weights": [
1.7543859649122806,
0.6993006993006993
],
"label_names": [
"benign",
"malignant"
]
}