retina_dev / config.json
steveGomez's picture
Upload config.json
d6fc232 verified
raw
history blame
961 Bytes
{
"model": {
"name": "Retinal Disease Classification Model",
"type": "Image to Text",
"input_shape": [224, 224, 3],
"output_shape": [1, "text"],
"layers": [
{
"type": "Conv2D",
"filters": 32,
"kernel_size": [3, 3],
"activation": "relu"
},
{
"type": "MaxPooling2D",
"pool_size": [2, 2]
},
{
"type": "Conv2D",
"filters": 64,
"kernel_size": [3, 3],
"activation": "relu"
},
{
"type": "MaxPooling2D",
"pool_size": [2, 2]
},
{
"type": "Flatten"
},
{
"type": "Dense",
"units": 128,
"activation": "relu"
},
{
"type": "Dense",
"units": 1,
"activation": "sigmoid"
}
],
"loss": "binary_crossentropy",
"optimizer": "adam",
"metrics": ["accuracy"]
}
}