config.json

{
    "input_size": 10,
    "hidden_size": 32,
    "output_size": 2,
    "num_samples": 1000,
    "batch_size": 32,
    "epochs": 20,
    "learning_rate": 0.001,
    "train_split": 0.8
}