{ "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 }