"for both training and testing."  (i would skip the "choose one of them"... it's kinda implicit in the previous sentence)