| { | |
| "architecture": "ResNet152", | |
| "description": "Custom classifier on top of ResNet152 for image classification", | |
| "input_shape": [224, 224, 3], | |
| "num_classes": 2, | |
| "optimizer": { | |
| "name": "Adam", | |
| "learning_rate": 0.001 | |
| }, | |
| "batch_size": 32, | |
| "epochs": 10, | |
| "loss_function": "binary_crossentropy", | |
| "metrics": ["accuracy"] | |
| } |