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