File size: 370 Bytes
713df86
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{
    "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"]
}