config.json

{
  "model_type": "custom_neural_network",
  "input_size": 7,
  "output_size": 3,
  "layers": [
    {
      "input": 7,
      "output": 4,
      "activation_function": "relu"
    },
    {
      "input": 4,
      "output": 3
    }
  ],
  "agent": {
    "learning_rate": 0.001,
    "loss_function": "MSELoss",
    "optimizer": "Adam"
  }
}