add tunable parameter to train and test ratio