| { | |
| "model_type": "CNN", | |
| "input_shape": [150, 150, 3], | |
| "num_classes": 4, | |
| "layers": [ | |
| { | |
| "type": "Conv2D", | |
| "filters": 32, | |
| "kernel_size": [3, 3], | |
| "activation": "relu" | |
| }, | |
| { | |
| "type": "Conv2D", | |
| "filters": 64, | |
| "kernel_size": [3, 3], | |
| "activation": "relu" | |
| }, | |
| { | |
| "type": "MaxPooling2D", | |
| "pool_size": [2, 2] | |
| }, | |
| { | |
| "type": "Dropout", | |
| "rate": 0.25 | |
| }, | |
| { | |
| "type": "Flatten" | |
| }, | |
| { | |
| "type": "Dense", | |
| "units": 128, | |
| "activation": "relu" | |
| }, | |
| { | |
| "type": "Dropout", | |
| "rate": 0.5 | |
| }, | |
| { | |
| "type": "Dense", | |
| "units": 4, | |
| "activation": "softmax" | |
| } | |
| ], | |
| "compile": { | |
| "loss": "categorical_crossentropy", | |
| "optimizer": "adam", | |
| "metrics": ["accuracy"] | |
| }, | |
| "training": { | |
| "batch_size": 64, | |
| "epochs": 10, | |
| "early_stopping": { | |
| "monitor": "val_loss", | |
| "patience": 5, | |
| "restore_best_weights": true | |
| } | |
| } | |
| } |