brain-mri-classifier / config.json
meghashyamreddy014's picture
Add model configuration
f6f9af0 verified
{
"model_type": "efficientnet",
"architecture": "efficientnet-b0",
"framework": "tensorflow",
"num_classes": 3,
"input_shape": [
150,
150,
3
],
"class_names": [
"CONTROL",
"AD",
"PD"
],
"class_labels": {
"0": "CONTROL - Normal Brain Scan",
"1": "AD - Alzheimer's Disease",
"2": "PD - Parkinson's Disease"
},
"preprocessing": {
"resize": [
150,
150
],
"normalize": "0-255",
"color_mode": "RGB"
},
"model_version": "1.0.0",
"training_info": {
"dataset": "Brain MRI scans for neurological disease detection",
"accuracy": "95.8%",
"validation_accuracy": "93.2%"
}
}