config.json

{
    "model_type": "Sequential",
    "input_size": [28, 28],
    "layers": [
        {
            "type": "Flatten",
            "input_shape": [28, 28]
        },
        {
            "type": "Dense",
            "units": 256,
            "activation": "relu"
        },
        {
            "type": "Dense",
            "units": 128,
            "activation": "relu"
        },
        {
            "type": "Dense",
            "units": 64,
            "activation": "relu"
        },
        {
            "type": "Dense",
            "units": 10,
            "activation": "softmax"
        }
    ],
    "optimizer": "adam",
    "loss_function": "sparse_categorical_crossentropy",
    "metrics": ["accuracy"]
}