PneumoniaDetect1 / config.json
ryefoxlime's picture
Update config.json
24c6059
raw
history blame contribute delete
386 Bytes
{
"model_type": "X-Ray_Classifiction",
"architectures": ["ResNet50V2"],
"num_classes": 2,
"class_names": ["NORMAL", "PNEUMONIA"],
"learning_rate": 0.000035,
"loss_function": "sparse_categorical_crossentropy",
"metric": "sparse_categorical_accuracy",
"imagenet_weights": true,
"trainable_base_model": false,
"normalize_input": true,
"resize_input": [224, 224, 3]
}