Partitioning data into training and validation datasets using R