Added hyperparameters for training and testing.