config.json

{
  "model_type": "ImprovedNet",
  "input_size": 109658,
  "output_size": 6,
  "hidden": [
    512,
    256,
    128
  ],
  "dropout": 0.3,
  "loss": "MSELoss",
  "optimizer": "Adam",
  "learning_rate": 0.0001,
  "num_epochs": 15
}