specified allowed parameters for training params instead of allowing all of them