config.json

{
    "architecture": "KerasSequential",
    "model_config": {
      "layers": [
        {
          "class_name": "ImageDataGenerator",
          "config": {
            "rotation_range": 20,
            "width_shift_range": 0.1,
            "height_shift_range": 0.1,
            "shear_range": 0.2,
            "zoom_range": 0.2,
            "horizontal_flip": true,
            "fill_mode": "nearest"
          }
        },
        {
          "class_name": "Rescaling",
          "config": {
            "scale": 0.00392156862745098
          }
        },
        {
          "class_name": "Conv2D",
          "config": {
            "filters": 16,
            "kernel_size": [3, 3],
            "padding": "same",
            "activation": "relu"
          }
        },
        {
          "class_name": "MaxPooling2D",
          "config": {
            "pool_size": [2, 2],
            "strides": null,
            "padding": "valid",
            "data_format": null
          }
        },
        {
          "class_name": "Conv2D",
          "config": {
            "filters": 32,
            "kernel_size": [3, 3],
            "padding": "same",
            "activation": "relu"
          }
        },
        {
          "class_name": "MaxPooling2D",
          "config": {
            "pool_size": [2, 2],
            "strides": null,
            "padding": "valid",
            "data_format": null
          }
        },
        {
          "class_name": "Conv2D",
          "config": {
            "filters": 64,
            "kernel_size": [3, 3],
            "padding": "same",
            "activation": "relu"
          }
        },
        {
          "class_name": "MaxPooling2D",
          "config": {
            "pool_size": [2, 2],
            "strides": null,
            "padding": "valid",
            "data_format": null
          }
        },
        {
          "class_name": "Dropout",
          "config": {
            "rate": 0.2,
            "noise_shape": null,
            "seed": null
          }
        },
        {
          "class_name": "Flatten",
          "config": {}
        },
        {
          "class_name": "Dense",
          "config": {
            "units": 128,
            "activation": "relu"
          }
        },
        {
          "class_name": "Dense",
          "config": {
            "units": 4,
            "activation": "softmax"
          }
        }
      ],
      "label_names": ["adidas", "newbalance", "nike", "puma"]
    }
  }