setTrainParams function

Check user input related to training