model_config.json

{
  "model_type": "neural_network",
  "architecture": {
    "layers": [
      {
        "type": "input",
        "shape": [None, 784]
      },
      {
        "type": "dense",
        "units": 128,
        "activation": "relu"
      },
      {
        "type": "dense",
        "units": 64,
        "activation": "relu"
      },
      {
        "type": "dense",
        "units": 10,
        "activation": "softmax"
      }
    ]
  },
  "training_config": {
    "optimizer": "adam",
    "loss": "categorical_crossentropy",
    "metrics": ["accuracy"]
  },
  "hyperparameters": {
    "learning_rate": 0.001,
    "batch_size": 32,
    "epochs": 50
  },
  "input_preprocessing": {
    "normalization": "min_max",
    "resize": [28, 28],
    "color_mode": "grayscale"
  },
  "metadata": {
    "created_at": "2024-01-01",
    "version": "1.0.0",
    "task": "classification"
  }
}