Training params update #34
Description
Right now the training params are defined in the params.py file. While these params are good defaults, they need to be able to be passed in when training is called (which is basically done already). We also need to use the function that calculates the optimal training params based on the input data