{ | |
"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] | |
} | |