config.json

{
  "input_features": 624,
  "learning_rate": 0.0001,
  "num_epochs": 20,
  "output_features": 2,
  "with_scheduler": true
}