Data partition between training test and validation sets